Prompts from file not working #379

1 files changed, 3 insertions, 4 deletions

diff --git a/scripts/prompts_from_file.py b/scripts/prompts_from_file.py
index da2ddd54..e0c7f1d2 100644
--- a/scripts/prompts_from_file.py
+++ b/scripts/prompts_from_file.py
@@ -25,17 +25,16 @@ class Script(scripts.Script):
         lines = [x for x in lines if len(x) > 0]

 

         batch_count = math.ceil(len(lines) / p.batch_size)

-        print(f"Will process {len(lines)} images in {batch_count} batches.")

+        print(f"Will process {len(lines) * p.n_iter} images in {batch_count * p.n_iter} batches.")

 

-        p.batch_count = 1

         p.do_not_save_grid = True

 

         state.job_count = batch_count

 

         images = []

         for batch_no in range(batch_count):

-            state.job = f"{batch_no} out of {batch_count}"

-            p.prompt = lines[batch_no*p.batch_size:(batch_no+1)*p.batch_size]

+            state.job = f"{batch_no} out of {batch_count * p.n_iter}"

+            p.prompt = lines[batch_no*p.batch_size:(batch_no+1)*p.batch_size] * p.n_iter

             proc = process_images(p)

             images += proc.images