From afc91edcb2e8c286f4c4fd758ec70700757c38d3 Mon Sep 17 00:00:00 2001
From: Yiximail <yiximail@qq.com>
Date: Sat, 23 Dec 2023 09:18:24 +0800
Subject: [PATCH] Reset the model_name after unloading the model (#5051)

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

diff --git a/modules/models.py b/modules/models.py
index 7a1124d1..e166f737 100644
--- a/modules/models.py
+++ b/modules/models.py
@@ -482,6 +482,7 @@ def clear_torch_cache():
 
 def unload_model():
     shared.model = shared.tokenizer = None
+    shared.model_name = 'None'
     shared.lora_names = []
     shared.model_dirty_from_training = False
     clear_torch_cache()