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

---
 modules/sd_models.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 1f43b2d9..c8d15e2a 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -586,6 +586,8 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
             return model_data.sd_model
 
         model_data.sd_model = None
+        model_management.unload_all_models()
+
         devices.torch_gc()
 
     timer.record("unload existing model")