oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-04 14:17:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a --torch-compile flag for transformers

Browse source
This commit is contained in:
oobabooga 2025-01-05 05:45:12 -08:00
parent 11af199aff
commit c0f600c887
5 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/ui.py
View file

@ -109,12 +109,13 @@ def list_model_elements():
'disk',
'cpu',
'bf16',
'load_in_4bit',
'load_in_8bit',
'torch_compile',
'trust_remote_code',
'no_use_fast',
'use_flash_attention_2',
'use_eager_attention',
'load_in_4bit',
'compute_dtype',
'quant_type',
'use_double_quant',