Commit 0af4127f authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

delete the field that is preventing the model from being unloaded and is...

delete the field that is preventing the model from being unloaded and is causing increased RAM usage
parent 02038036
Loading
Loading
Loading
Loading
+4 −0
Original line number Diff line number Diff line
@@ -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