diff --git a/modules/sd_vae_consistency.py b/modules/sd_vae_consistency.py index cf189a105..e700e538e 100644 --- a/modules/sd_vae_consistency.py +++ b/modules/sd_vae_consistency.py @@ -30,4 +30,5 @@ def decoder_model(): def unload(): global sd_vae_consistency_models if sd_vae_consistency_models is not None: + devices.torch_gc() sd_vae_consistency_models.ckpt.to('cpu')