Simplify a function

2023-04-07 21:37:41 -03:00
parent ffd102e5c0
commit 0b458bf82d
1 changed files with 1 additions and 2 deletions
--- a/modules/models.py
+++ b/modules/models.py
@@ -194,8 +194,7 @@ def unload_model():


 def reload_model():
-    shared.model = shared.tokenizer = None
-    clear_torch_cache()
+    unload_model()
    shared.model, shared.tokenizer = load_model(shared.model_name)