From b153ebe3b679854e7314a6d33a038f9f0cc0c233 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Fri, 16 Sep 2022 08:33:47 +0300
Subject: fix loopback color correction to store color_correction info for
 initial image and use that instead of extracting from previous image in a
 loop #481 #541

---
 scripts/loopback.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'scripts/loopback.py')

diff --git a/scripts/loopback.py b/scripts/loopback.py
index f1102c92..8aca61f3 100644
--- a/scripts/loopback.py
+++ b/scripts/loopback.py
@@ -40,6 +40,9 @@ 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])]
+
         for n in range(batch_count):
             history = []
 
-- 
cgit v1.2.3