diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-18 04:31:01 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-09-18 04:31:01 +0000 |
| commit | e9a9764e002158e943ca519da4e7e476cc50810d (patch) | |
| tree | ed35e82b2c797495ca4dbccc7921080181a838a9 /scripts/loopback.py | |
| parent | c57ae054e9fafcea0567c4a00765b805dbc38ec3 (diff) | |
| parent | 2dbc9c5903cd2788f4ef21fe8ebfdd8e62f642cb (diff) | |
| download | stable-diffusion-webui-gfx803-e9a9764e002158e943ca519da4e7e476cc50810d.tar.gz stable-diffusion-webui-gfx803-e9a9764e002158e943ca519da4e7e476cc50810d.tar.bz2 stable-diffusion-webui-gfx803-e9a9764e002158e943ca519da4e7e476cc50810d.zip | |
Merge pull request #625 from bdrupieski/fix-batch-processing-color-correction
fix batch color correction
Diffstat (limited to 'scripts/loopback.py')
| -rw-r--r-- | scripts/loopback.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/scripts/loopback.py b/scripts/loopback.py index 8aca61f3..e90b58d4 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 = []
@@ -51,6 +50,9 @@ class Script(scripts.Script): p.batch_size = 1
p.do_not_save_grid = True
+ if opts.img2img_color_correction:
+ p.color_corrections = initial_color_corrections
+
state.job = f"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}"
processed = processing.process_images(p)
|