mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-09-20 10:35:10 +02:00
Reset the model_name after unloading the model (#5051)
This commit is contained in:
parent
c1b99f45cb
commit
afc91edcb2
1 changed files with 1 additions and 0 deletions
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue