AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Minor fix

Browse Source
This commit is contained in:
oobabooga 2024-04-05 18:43:43 -07:00
parent 6b7f7555fc
commit 1b87844928
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/models.py
View File

@ -137,7 +137,7 @@ def huggingface_loader(model_name):
params['use_flash_attention_2'] = True params['use_flash_attention_2'] = True
if shared.args.force_safetensors: 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) config = AutoConfig.from_pretrained(path_to_model, trust_remote_code=shared.args.trust_remote_code)