|
|
|
|
@ -50,9 +50,9 @@ def apply_color_correction(correction, original_image):
|
|
|
|
|
correction,
|
|
|
|
|
channel_axis=2
|
|
|
|
|
), cv2.COLOR_LAB2RGB).astype("uint8"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
image = blendLayers(image, original_image, BlendType.LUMINOSITY)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return image
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -470,6 +470,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
|
|
|
|
|
if k == 'sd_model_checkpoint': sd_models.reload_model_weights() # make onchange call for changing SD model
|
|
|
|
|
if k == 'sd_vae': sd_vae.reload_vae_weights() # make onchange call for changing VAE
|
|
|
|
|
|
|
|
|
|
# Assign sd_model here to ensure that it reflects the model after any changes
|
|
|
|
|
p.sd_model = shared.sd_model
|
|
|
|
|
res = process_images_inner(p)
|
|
|
|
|
|
|
|
|
|
finally:
|
|
|
|
|
|