fix: save images from ui when a batch was run

When doing batches, it was not possible to save images from the ui because the prompts and images sent to the ui didn't align.
2023-08-01 18:06:25 +02:00
parent ee1fec03ff
commit 43509333ec
1 changed files with 5 additions and 0 deletions
@@ -1062,9 +1062,14 @@ class Script(scripts.Script):
            processed = process_images(p)
            images += processed.images
            infotexts += processed.infotexts
+            
+            # prompt and seed info for batch grid
            all_seeds.append(processed.seed)
            all_prompts.append(processed.prompt)
            
+            # prompt and seed info for individual images
+            all_seeds += processed.all_seeds
+            all_prompts += processed.all_prompts
            
            state.job = f"{state.job_no} out of {state.job_count}"