diff options
| author | Brian Drupieski <brian.drupieski@gmail.com> | 2022-09-17 22:18:30 +0000 |
|---|---|---|
| committer | Brian Drupieski <brian.drupieski@gmail.com> | 2022-09-17 22:18:30 +0000 |
| commit | 7bc5739fe27809802ed24d1833b0ef443ec64aa9 (patch) | |
| tree | 93de0170e18da0d044151c7fc6b642480a0220f6 /scripts/loopback.py | |
| parent | 663af7c1c8bb5aa8ccd6bb1634b51f7ed7a32740 (diff) | |
| download | stable-diffusion-webui-gfx803-7bc5739fe27809802ed24d1833b0ef443ec64aa9.tar.gz stable-diffusion-webui-gfx803-7bc5739fe27809802ed24d1833b0ef443ec64aa9.tar.bz2 stable-diffusion-webui-gfx803-7bc5739fe27809802ed24d1833b0ef443ec64aa9.zip | |
set loopback color corrections on each iteration
Diffstat (limited to 'scripts/loopback.py')
| -rw-r--r-- | scripts/loopback.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scripts/loopback.py b/scripts/loopback.py index 8aca61f3..2a3f4d6d 100644 --- a/scripts/loopback.py +++ b/scripts/loopback.py @@ -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}"
|