admin/FiDA-Gen-Img-Flux2-klein
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
zcr
2026-03-24 16:08:21 +08:00
parent a4bdad2872
commit 36dc3d08a1
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/litserve_serve.py
View File

@@ -97,6 +97,8 @@ class FluxKleinAPI(ls.LitAPI):
images = payload.get("images", []) images = payload.get("images", [])
prompt = payload.get("prompt", "") prompt = payload.get("prompt", "")
gen = torch.Generator(device=self.device) gen = torch.Generator(device=self.device)
seed = gen.seed()
print(f"本次使用的随机种子是: {seed}")
if images: if images:
output = self.pipe( output = self.pipe(
image=images, image=images,