set the field to None instead
parent
0af4127fd1
commit
dca121e903
|
|
@ -1076,13 +1076,13 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
|
||||||
current = shared.sd_model.sd_checkpoint_info
|
current = shared.sd_model.sd_checkpoint_info
|
||||||
try:
|
try:
|
||||||
if self.hr_checkpoint_info is not None:
|
if self.hr_checkpoint_info is not None:
|
||||||
del self.sampler
|
self.sampler = None
|
||||||
sd_models.reload_model_weights(info=self.hr_checkpoint_info)
|
sd_models.reload_model_weights(info=self.hr_checkpoint_info)
|
||||||
devices.torch_gc()
|
devices.torch_gc()
|
||||||
|
|
||||||
return self.sample_hr_pass(samples, decoded_samples, seeds, subseeds, subseed_strength, prompts)
|
return self.sample_hr_pass(samples, decoded_samples, seeds, subseeds, subseed_strength, prompts)
|
||||||
finally:
|
finally:
|
||||||
del self.sampler
|
self.sampler = None
|
||||||
sd_models.reload_model_weights(info=current)
|
sd_models.reload_model_weights(info=current)
|
||||||
devices.torch_gc()
|
devices.torch_gc()
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue