delete the field that is preventing the model from being unloaded and is causing increased RAM usage

1 files changed, 4 insertions, 0 deletions

diff --git a/modules/processing.py b/modules/processing.py
index 0677de81..b09433b0 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -1076,11 +1076,15 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
         current = shared.sd_model.sd_checkpoint_info

         try:

             if self.hr_checkpoint_info is not None:

+                del self.sampler

                 sd_models.reload_model_weights(info=self.hr_checkpoint_info)

+                devices.torch_gc()

 

             return self.sample_hr_pass(samples, decoded_samples, seeds, subseeds, subseed_strength, prompts)

         finally:

+            del self.sampler

             sd_models.reload_model_weights(info=current)

+            devices.torch_gc()

 

     def sample_hr_pass(self, samples, decoded_samples, seeds, subseeds, subseed_strength, prompts):

         self.is_hr_pass = True