From 640cb1bb8dfbd05a3927753090bc80fb5b9737b1 Mon Sep 17 00:00:00 2001 From: w-e-w <40751091+w-e-w@users.noreply.github.com> Date: Fri, 18 Aug 2023 17:14:02 +0900 Subject: [PATCH] fix model override logic do not need extra logic to unload refine model --- modules/processing.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/modules/processing.py b/modules/processing.py index e62db62fd..263baefc1 100755 --- a/modules/processing.py +++ b/modules/processing.py @@ -702,11 +702,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed: stored_opts = {k: opts.data[k] for k in p.override_settings.keys()} try: - # after running refiner, the refiner model is not unloaded - webui swaps back to main model here - if shared.sd_model.sd_checkpoint_info.title != opts.sd_model_checkpoint: - sd_models.reload_model_weights() - # if no checkpoint override or the override checkpoint can't be found, remove override entry and load opts checkpoint + # and if after running refiner, the refiner model is not unloaded - webui swaps back to main model here, if model over is present it will be reloaded afterwards if sd_models.checkpoint_aliases.get(p.override_settings.get('sd_model_checkpoint')) is None: p.override_settings.pop('sd_model_checkpoint', None) sd_models.reload_model_weights()