mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-09-20 18:45:09 +02:00
Minor fix
This commit is contained in:
parent
6b7f7555fc
commit
1b87844928
1 changed files with 1 additions and 1 deletions
|
@ -137,7 +137,7 @@ def huggingface_loader(model_name):
|
|||
params['use_flash_attention_2'] = True
|
||||
|
||||
if shared.args.force_safetensors:
|
||||
params['max_memory'] = True
|
||||
params['force_safetensors'] = True
|
||||
|
||||
config = AutoConfig.from_pretrained(path_to_model, trust_remote_code=shared.args.trust_remote_code)
|
||||
|
||||
|
|
Loading…
Reference in a new issue