StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Prompts from file not working #379

Browse Source
This commit is contained in:
AUTOMATIC 2022-09-13 09:41:38 +03:00
parent 9ea44c7ce7
commit afbb3504da
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/prompts_from_file.py
View File

@ -25,17 +25,16 @@ class Script(scripts.Script):
lines = [x for x in lines if len(x) > 0] lines = [x for x in lines if len(x) > 0]
batch_count = math.ceil(len(lines) / p.batch_size) 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 p.do_not_save_grid = True
state.job_count = batch_count state.job_count = batch_count
images = [] images = []
for batch_no in range(batch_count): for batch_no in range(batch_count):
state.job = f"{batch_no} out of {batch_count}" 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.prompt = lines[batch_no*p.batch_size:(batch_no+1)*p.batch_size] * p.n_iter
proc = process_images(p) proc = process_images(p)
images += proc.images images += proc.images