Update rp_handler.py

runpod-workers · Oct 3, 2023 · f65d4f4 · f65d4f4
1 parent f2c8618
commit f65d4f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rp_handler.py b/src/rp_handler.py
@@ -107,7 +107,7 @@ def generate_image(job):
             strength=job_input['strength'],
             image=init_image,
             generator=generator
-        ).images[0]
+        ).images
     else:
         # Generate latent image using pipe
         image = base(