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