From a36a30fb933723d983edd90044a697fe75bfb3b0 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 7 Nov 2023 13:01:10 +0800
Subject: [PATCH] add gc after using consistency dec

---
 modules/sd_vae_consistency.py | 1 +
 1 file changed, 1 insertion(+)

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')