add gc after using consistency dec

2023-11-07 13:01:10 +08:00 · 2023-11-07 13:01:10 +08:00 · a36a30fb93
parent 2ea8726597
commit a36a30fb93
1 changed files with 1 additions and 0 deletions
--- 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')