aboutsummaryrefslogtreecommitdiffstats
path: root/scripts/batch.py
diff options
context:
space:
mode:
authorBrian Drupieski <brian.drupieski@gmail.com>2022-09-17 22:18:30 +0000
committerBrian Drupieski <brian.drupieski@gmail.com>2022-09-17 22:18:30 +0000
commit7bc5739fe27809802ed24d1833b0ef443ec64aa9 (patch)
tree93de0170e18da0d044151c7fc6b642480a0220f6 /scripts/batch.py
parent663af7c1c8bb5aa8ccd6bb1634b51f7ed7a32740 (diff)
downloadstable-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/batch.py')
-rw-r--r--scripts/batch.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/scripts/batch.py b/scripts/batch.py
index fc9e6b5f..a117fcbf 100644
--- a/scripts/batch.py
+++ b/scripts/batch.py
@@ -6,7 +6,7 @@ import traceback
import modules.scripts as scripts
import gradio as gr
-from modules.processing import Processed, process_images, setup_color_correction
+from modules.processing import Processed, process_images
from PIL import Image
from modules.shared import opts, cmd_opts, state
@@ -52,9 +52,6 @@ class Script(scripts.Script):
state.job = f"{batch_no} out of {batch_count}: {batch_images[0][1]}"
p.init_images = [x[0] for x in batch_images]
- if opts.img2img_color_correction:
- p.color_corrections = [setup_color_correction(i) for i in p.init_images]
-
proc = process_images(p)
for image, (_, path) in zip(proc.images, batch_images):
filename = os.path.basename(path)