From f65d4f441a430975486f26ed980162f9aa042ca2 Mon Sep 17 00:00:00 2001
From: Justin Merrell <merrelljustin@gmail.com>
Date: Tue, 3 Oct 2023 08:20:08 -0400
Subject: [PATCH] Update rp_handler.py

---
 src/rp_handler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/rp_handler.py b/src/rp_handler.py
index 7da5c47..98450cf 100644
--- 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(