hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-03 10:50:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

set loopback color corrections on each iteration

Browse Source
This commit is contained in:
Brian Drupieski
2022-09-17 18:18:30 -04:00
parent 663af7c1c8
commit 7bc5739fe2
3 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/loopback.py
View File

@@ -40,8 +40,7 @@ class Script(scripts.Script):
all_images = []
state.job_count = loops * batch_count
if opts.img2img_color_correction:
p.color_corrections = [processing.setup_color_correction(p.init_images[0])]
initial_color_corrections = [processing.setup_color_correction(p.init_images[0])]
for n in range(batch_count):
history = []
@@ -50,6 +49,7 @@ class Script(scripts.Script):
p.n_iter = 1
p.batch_size = 1
p.do_not_save_grid = True
p.color_corrections = initial_color_corrections
state.job = f"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}"