only set loopback color corrections if corrections enabled

2dbc9c59 · Brian Drupieski · cf70528b · 2dbc9c59
Commit 2dbc9c59 authored Sep 17, 2022 by Brian Drupieski
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

scripts/loopback.py scripts/loopback.py +3 -1

No files found.
--- a/scripts/loopback.py
+++ b/scripts/loopback.py
@@ -49,7 +49,9 @@ class Script(scripts.Script):
                p.n_iter = 1
                p.batch_size = 1
                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}"