Commit 0af4127f authored by AUTOMATIC1111's avatar AUTOMATIC1111

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
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment