diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-05 07:11:52 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-05 07:11:52 +0000 |
| commit | 6745e8c5f2b01932a354c4977b4a9eaf488a224b (patch) | |
| tree | cbadd09c9305099db7763532b7aaf909d366d326 /modules/processing.py | |
| parent | 5f4fa942b8ec3ed3b15a352903489d6f9e6eb46e (diff) | |
| parent | 83ca8dd0c96e3cc8dd444e8d980c07718dc647ee (diff) | |
| download | stable-diffusion-webui-gfx803-6745e8c5f2b01932a354c4977b4a9eaf488a224b.tar.gz stable-diffusion-webui-gfx803-6745e8c5f2b01932a354c4977b4a9eaf488a224b.tar.bz2 stable-diffusion-webui-gfx803-6745e8c5f2b01932a354c4977b4a9eaf488a224b.zip | |
Merge pull request #6349 from philpax/fix-sd-arch-switch-in-override-settings
fix(api): assign sd_model after settings change (v2)
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/processing.py b/modules/processing.py index 7e853287..a12bd9e8 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
@@ -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:
|