1

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,