fix(api): assign sd_model after settings change

1 files changed, 4 insertions, 2 deletions

diff --git a/modules/processing.py b/modules/processing.py
index 4a406084..0b270278 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -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

 

 

@@ -466,6 +466,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: