[Bugfix] fix lora_dtype value type in arg_utils.py (vllm-project#5398)

xjpang · Jun 27, 2024 · 79fc9ca · 79fc9ca
1 parent 7698322
commit 79fc9ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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