initial refiner support

1 files changed, 4 insertions, 0 deletions

diff --git a/modules/processing.py b/modules/processing.py
index 31745006..f4748d6d 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -666,6 +666,10 @@ 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

         if sd_models.checkpoint_aliases.get(p.override_settings.get('sd_model_checkpoint')) is None:

             p.override_settings.pop('sd_model_checkpoint', None)