set the field to None instead

This commit is contained in:
AUTOMATIC1111 2023-07-31 09:13:07 +03:00
parent 0af4127fd1
commit dca121e903

View File

@ -1076,13 +1076,13 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
current = shared.sd_model.sd_checkpoint_info
try:
if self.hr_checkpoint_info is not None:
del self.sampler
self.sampler = None
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
self.sampler = None
sd_models.reload_model_weights(info=current)
devices.torch_gc()