only set loopback color corrections if corrections enabled

This commit is contained in:
Brian Drupieski 2022-09-17 18:20:43 -04:00
parent cf70528bce
commit 2dbc9c5903
1 changed files with 3 additions and 1 deletions

View File

@ -49,7 +49,9 @@ class Script(scripts.Script):
p.n_iter = 1 p.n_iter = 1
p.batch_size = 1 p.batch_size = 1
p.do_not_save_grid = True p.do_not_save_grid = True
p.color_corrections = initial_color_corrections
if opts.img2img_color_correction:
p.color_corrections = initial_color_corrections
state.job = f"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}" state.job = f"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}"