Commit
·
ad31d2d
1
Parent(s):
ecfb7d9
Refactor flux_helpers.py to fix loading of Lora weights
Browse files
modules/helpers/flux_helpers.py
CHANGED
@@ -96,7 +96,7 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
|
|
96 |
'num_images_per_prompt': request.num_images_per_prompt,
|
97 |
'num_inference_steps': request.num_inference_steps,
|
98 |
'guidance_scale': request.guidance_scale,
|
99 |
-
'generator': [torch.Generator(
|
100 |
}
|
101 |
|
102 |
if request.controlnet_config:
|
|
|
96 |
'num_images_per_prompt': request.num_images_per_prompt,
|
97 |
'num_inference_steps': request.num_inference_steps,
|
98 |
'guidance_scale': request.guidance_scale,
|
99 |
+
'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)],
|
100 |
}
|
101 |
|
102 |
if request.controlnet_config:
|