This commit is contained in:
oobabooga 2025-06-01 19:27:14 -07:00
parent 92adceb7b5
commit ad6d0218ae

View file

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