Commit ·
e7eda15
1
Parent(s): 3d75865
Refactor generator initialization in image_tab function
Browse files- tabs/images/handlers.py +1 -1
tabs/images/handlers.py
CHANGED
|
@@ -227,7 +227,7 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq, progress=gr.Prog
|
|
| 227 |
'num_images_per_prompt': request.num_images_per_prompt,
|
| 228 |
'num_inference_steps': request.num_inference_steps,
|
| 229 |
'guidance_scale': request.guidance_scale,
|
| 230 |
-
'generator': [torch.Generator(
|
| 231 |
}
|
| 232 |
|
| 233 |
if isinstance(pipeline, sd_pipes):
|
|
|
|
| 227 |
'num_images_per_prompt': request.num_images_per_prompt,
|
| 228 |
'num_inference_steps': request.num_inference_steps,
|
| 229 |
'guidance_scale': request.guidance_scale,
|
| 230 |
+
'generator': [torch.Generator().manual_seed(request.seed + i) if not request.seed is any([None, 0, -1]) else torch.Generator().manual_seed(random.randint(0, 2**32 - 1)) for i in range(request.num_images_per_prompt)],
|
| 231 |
}
|
| 232 |
|
| 233 |
if isinstance(pipeline, sd_pipes):
|