diff options
author | Brian Drupieski <brian.drupieski@gmail.com> | 2022-09-17 22:18:30 +0000 |
---|---|---|
committer | Brian Drupieski <brian.drupieski@gmail.com> | 2022-09-17 22:18:30 +0000 |
commit | 7bc5739fe27809802ed24d1833b0ef443ec64aa9 (patch) | |
tree | 93de0170e18da0d044151c7fc6b642480a0220f6 /scripts/batch.py | |
parent | 663af7c1c8bb5aa8ccd6bb1634b51f7ed7a32740 (diff) | |
download | stable-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.py | 5 |
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)
|