add gc after using consistency dec
This commit is contained in:
parent
2ea8726597
commit
a36a30fb93
|
@ -30,4 +30,5 @@ def decoder_model():
|
||||||
def unload():
|
def unload():
|
||||||
global sd_vae_consistency_models
|
global sd_vae_consistency_models
|
||||||
if sd_vae_consistency_models is not None:
|
if sd_vae_consistency_models is not None:
|
||||||
|
devices.torch_gc()
|
||||||
sd_vae_consistency_models.ckpt.to('cpu')
|
sd_vae_consistency_models.ckpt.to('cpu')
|
||||||
|
|
Loading…
Reference in New Issue