Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Simplify a function

Browse Source
This commit is contained in:
oobabooga 2023-04-07 21:37:41 -03:00
parent ffd102e5c0
commit 0b458bf82d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/models.py
View File

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