Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 01:09:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
oobabooga 2024-07-04 20:16:44 -07:00
parent a47de06088
commit 8b44d7b12a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/models.py
View File

@ -147,7 +147,7 @@ def huggingface_loader(model_name):
params['force_safetensors'] = True params['force_safetensors'] = True
if shared.args.use_eager_attention: if shared.args.use_eager_attention:
params['attn_implementation'] = 'eager' params['attn_implementation'] = 'eager'
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)