From f993ab1c2cfdc1f9d1b06f432e883def463759d5 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 25 Jan 2024 07:15:51 -0800
Subject: [PATCH] Update sd_models.py

---
 modules/sd_models.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/modules/sd_models.py b/modules/sd_models.py
index c8d15e2a..942cd7c4 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -587,8 +587,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
 
         model_data.sd_model = None
         model_management.unload_all_models()
-
-        devices.torch_gc()
+        model_management.soft_empty_cache()
 
     timer.record("unload existing model")