Produce 4 upscaled images

Signed-off-by: bghira <[email protected]>
bghira · May 7, 2023 · 3381a2a · 3381a2a
1 parent b22680d
commit 3381a2a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -281,10 +281,11 @@ def _run_pipeline(
             elif upscaler:
                 new_image = pipe(
                     prompt_embeds=prompt_embed,
+                    num_images_per_prompt=4,
                     negative_prompt_embeds=negative_embed,
                     image=image,
                     num_inference_steps=int(float(steps)),
-                ).images[0]
+                ).images
             else:
                 raise Exception(
                     "Invalid combination of parameters for image generation"