From 2217fe419d2df0bd7fe7baa65675643f199a4681 Mon Sep 17 00:00:00 2001 From: AIrjen Date: Sun, 23 Apr 2023 08:06:59 +0200 Subject: [PATCH] Added all_seeds and all_prompts to processing logic --- scripts/onebuttonprompt.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/onebuttonprompt.py b/scripts/onebuttonprompt.py index c9d9ad4..a9da315 100644 --- a/scripts/onebuttonprompt.py +++ b/scripts/onebuttonprompt.py @@ -254,6 +254,8 @@ class Script(scripts.Script): images = [] infotexts = [] + all_seeds = [] + all_prompts = [] batches = p.n_iter initialbatchsize = p.batch_size @@ -262,6 +264,7 @@ class Script(scripts.Script): p.batch_size = 1 originalprompt = p.prompt + if(silentmode and workprompt != ""): print("Workflow mode turned on, not generating a prompt. Using workflow prompt.") elif(silentmode): @@ -333,6 +336,8 @@ class Script(scripts.Script): 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): @@ -342,4 +347,4 @@ class Script(scripts.Script): # just return all the things p.n_iter = 0 p.batch_size = 0 - return Processed(p=p, images_list=images, info=infotexts[0], infotexts=infotexts) + return Processed(p=p, images_list=images, info=infotexts[0], infotexts=infotexts, all_seeds=all_seeds, all_prompts=all_prompts)