From 8bd1129c7a435b2ca25349c5b63c03a829173307 Mon Sep 17 00:00:00 2001
From: AIrjen <130234949+AIrjen@users.noreply.github.com>
Date: Sat, 13 May 2023 20:36:12 +0200
Subject: [PATCH] Fix for batch sizes (#26)

---
 scripts/onebuttonprompt.py | 36 ++++++++++++++++++++++--------------
 1 file changed, 22 insertions(+), 14 deletions(-)

diff --git a/scripts/onebuttonprompt.py b/scripts/onebuttonprompt.py
index a330a82..7d8bd32 100644
--- a/scripts/onebuttonprompt.py
+++ b/scripts/onebuttonprompt.py
@@ -656,21 +656,29 @@ class Script(scripts.Script):
             if(silentmode == True):
                 p.prompt = workprompt
 
-            for j in range(batchsize):
+            #for j in range(batchsize):
        
-                print(" ")
-                print("Full prompt to be processed:")
-                print(" ")
-                print(p.prompt)
-                processed = process_images(p)
-                images += processed.images
-                infotexts += processed.infotexts
-                all_seeds.append(processed.seed)
-                all_prompts.append(processed.prompt)
-            
-                # Only move up a seed, when there are multiple batchsizes, and we had the first one done.
-                if(initialbatchsize != 1):
-                    p.seed += 1
+            print(" ")
+            print("Full prompt to be processed:")
+            print(" ")
+            print(p.prompt)
+
+            # finally figured out how to do multiple batch sizes
+            promptlist = []
+            for i in range(batchsize):
+                promptlist.append(p.prompt)
+
+            p.prompt = promptlist
+            p.batch_size = batchsize
+            processed = process_images(p)
+            images += processed.images
+            infotexts += processed.infotexts
+            all_seeds.append(processed.seed)
+            all_prompts.append(processed.prompt)
+        
+            # Only move up a seed, when there are multiple batchsizes, and we had the first one done.
+            if(initialbatchsize != 1):
+                p.seed += 1
             
             p.seed +=1    
         # just return all the things