mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2025-06-07 14:17:09 -04:00
Fix after 219f0a7731
This commit is contained in:
parent
92adceb7b5
commit
ad6d0218ae
1 changed files with 1 additions and 1 deletions
|
@ -116,7 +116,7 @@ def unload_model(keep_model_name=False):
|
||||||
return
|
return
|
||||||
|
|
||||||
is_llamacpp = (shared.model.__class__.__name__ == 'LlamaServer')
|
is_llamacpp = (shared.model.__class__.__name__ == 'LlamaServer')
|
||||||
if shared.args.loader == 'ExLlamav3_HF':
|
if shared.model.__class__.__name__ == 'Exllamav3HF':
|
||||||
shared.model.unload()
|
shared.model.unload()
|
||||||
|
|
||||||
shared.model = shared.tokenizer = None
|
shared.model = shared.tokenizer = None
|
||||||
|
|
Loading…
Add table
Reference in a new issue