diff --git a/modules/models.py b/modules/models.py index 6fa7dbbd..75dff911 100644 --- 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)