openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-15 15:13:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6349 from philpax/fix-sd-arch-switch-in-override-settings

Browse Source 
fix(api): assign sd_model after settings change (v2)
This commit is contained in:
AUTOMATIC1111 2023-01-05 10:11:52 +03:00 committed by GitHub
commit 6745e8c5f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/processing.py
View File

@ -50,9 +50,9 @@ def apply_color_correction(correction, original_image):
correction, correction,
channel_axis=2 channel_axis=2
), cv2.COLOR_LAB2RGB).astype("uint8")) ), cv2.COLOR_LAB2RGB).astype("uint8"))
image = blendLayers(image, original_image, BlendType.LUMINOSITY) image = blendLayers(image, original_image, BlendType.LUMINOSITY)
return image 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_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 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) res = process_images_inner(p)
finally: finally: