Commit 2e309974 authored by AUTOMATIC's avatar AUTOMATIC

move sd_model assignment to the place where we change the sd_model

parent f3df2615
...@@ -466,12 +466,16 @@ def process_images(p: StableDiffusionProcessing) -> Processed: ...@@ -466,12 +466,16 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
try: try:
for k, v in p.override_settings.items(): for k, v in p.override_settings.items():
setattr(opts, k, v) setattr(opts, k, v)
if k == 'sd_hypernetwork': shared.reload_hypernetworks() # make onchange call for changing hypernet if k == 'sd_hypernetwork':
if k == 'sd_model_checkpoint': sd_models.reload_model_weights() # make onchange call for changing SD model shared.reload_hypernetworks() # make onchange call for changing hypernet
if k == 'sd_vae': sd_vae.reload_vae_weights() # make onchange call for changing VAE
if k == 'sd_model_checkpoint':
sd_models.reload_model_weights() # make onchange call for changing SD model
p.sd_model = shared.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) res = process_images_inner(p)
finally: finally:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment