Commit fa931733 authored by Philpax's avatar Philpax
Browse files

fix(api): assign sd_model after settings change

parent c6f347b8

modules/api/api.py

+0 −2
Original line number Diff line number Diff line
@@ -121,7 +121,6 @@ class Api: 


    def text2imgapi(self, txt2imgreq: StableDiffusionTxt2ImgProcessingAPI):

        populate = txt2imgreq.copy(update={ # Override __init__ params

            "sd_model": shared.sd_model,

            "sampler_name": validate_sampler_name(txt2imgreq.sampler_name or txt2imgreq.sampler_index),

            "do_not_save_samples": True,

            "do_not_save_grid": True
@@ -153,7 +152,6 @@ class Api: 
            mask = decode_base64_to_image(mask)



        populate = img2imgreq.copy(update={ # Override __init__ params

            "sd_model": shared.sd_model,

            "sampler_name": validate_sampler_name(img2imgreq.sampler_name or img2imgreq.sampler_index),

            "do_not_save_samples": True,

            "do_not_save_grid": True,

modules/processing.py

+4 −2
Original line number Diff line number Diff line
@@ -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:

CRA Git | Maintained and supported by SUSTech CRA and CCSE