diff --git a/app/litserve_serve.py b/app/litserve_serve.py index b302e98..f5f8814 100755 --- a/app/litserve_serve.py +++ b/app/litserve_serve.py @@ -97,6 +97,8 @@ class FluxKleinAPI(ls.LitAPI): images = payload.get("images", []) prompt = payload.get("prompt", "") gen = torch.Generator(device=self.device) + seed = gen.seed() + print(f"本次使用的随机种子是: {seed}") if images: output = self.pipe( image=images,