mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2025-06-07 06:06:20 -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
|
||||
|
||||
is_llamacpp = (shared.model.__class__.__name__ == 'LlamaServer')
|
||||
if shared.args.loader == 'ExLlamav3_HF':
|
||||
if shared.model.__class__.__name__ == 'Exllamav3HF':
|
||||
shared.model.unload()
|
||||
|
||||
shared.model = shared.tokenizer = None
|
||||
|
|
Loading…
Add table
Reference in a new issue