squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Doc] Remove float32 choice from --lora-dtype (#10348)

Browse Source 
Signed-off-by: Xin Yang <xyang19@gmail.com>
This commit is contained in:
Xin Yang 2024-11-15 02:22:57 -08:00 committed by GitHub
parent b311efd0bd
commit 26908554b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/engine/arg_utils.py
View File

@ -601,7 +601,7 @@ class EngineArgs:
'--lora-dtype', '--lora-dtype',
type=str, type=str,
default=EngineArgs.lora_dtype, default=EngineArgs.lora_dtype,
choices=['auto', 'float16', 'bfloat16', 'float32'], choices=['auto', 'float16', 'bfloat16'],
help=('Data type for LoRA. If auto, will default to ' help=('Data type for LoRA. If auto, will default to '
'base model dtype.')) 'base model dtype.'))
parser.add_argument( parser.add_argument(