michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix batch color correction

Browse source
This commit is contained in:
Brian Drupieski 2022-09-17 13:57:01 -04:00
parent 23a0ec04c0
commit 663af7c1c8
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
scripts/batch.py
View file

@ -6,7 +6,7 @@ import traceback
import modules.scripts as scripts import modules.scripts as scripts
import gradio as gr import gradio as gr
from modules.processing import Processed, process_images from modules.processing import Processed, process_images, setup_color_correction
from PIL import Image from PIL import Image
from modules.shared import opts, cmd_opts, state from modules.shared import opts, cmd_opts, state
@ -51,6 +51,10 @@ class Script(scripts.Script):
state.job = f"{batch_no} out of {batch_count}: {batch_images[0][1]}" state.job = f"{batch_no} out of {batch_count}: {batch_images[0][1]}"
p.init_images = [x[0] for x in batch_images] 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) proc = process_images(p)
for image, (_, path) in zip(proc.images, batch_images): for image, (_, path) in zip(proc.images, batch_images):
filename = os.path.basename(path) filename = os.path.basename(path)