Spaces:
Running
Running
:tada: change some process
Browse files- src/inference.py +1 -1
src/inference.py
CHANGED
@@ -106,7 +106,7 @@ def inference(
|
|
106 |
height=height,
|
107 |
guidance_scale=guidance_scale,
|
108 |
num_inference_steps=num_inference_steps,
|
109 |
-
|
110 |
generator=generator,
|
111 |
).images
|
112 |
|
|
|
106 |
height=height,
|
107 |
guidance_scale=guidance_scale,
|
108 |
num_inference_steps=num_inference_steps,
|
109 |
+
num_images_per_prompt=num_images,
|
110 |
generator=generator,
|
111 |
).images
|
112 |
|