mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-01-07 17:20:19 +01:00
Fix after 219f0a7731
This commit is contained in:
parent
92adceb7b5
commit
ad6d0218ae
|
|
@ -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…
Reference in a new issue