From 3fe019daedb03eaad885730f43f1a12c71b0371a Mon Sep 17 00:00:00 2001 From: Ali Panahi <64020589+c3-ali@users.noreply.github.com> Date: Mon, 10 Jun 2024 17:00:27 -0700 Subject: [PATCH] fix lora_dtype value type in arg_utils.py --- vllm/engine/arg_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/engine/arg_utils.py b/vllm/engine/arg_utils.py index b7e815db12eb..6317b18ff87e 100644 --- a/vllm/engine/arg_utils.py +++ b/vllm/engine/arg_utils.py @@ -68,7 +68,7 @@ class EngineArgs: fully_sharded_loras: bool = False lora_extra_vocab_size: int = 256 long_lora_scaling_factors: Optional[Tuple[float]] = None - lora_dtype = 'auto' + lora_dtype: str = 'auto' max_cpu_loras: Optional[int] = None device: str = 'auto' ray_workers_use_nsight: bool = False