feat: pass the `max_lora_rank` parameter to vLLM backend

This commit is contained in:
juejuezi 2024-05-17 16:07:39 +08:00
parent 8af9817605
commit b20d62ba3c
2 changed files with 2 additions and 0 deletions

View File

@ -59,6 +59,7 @@ class VllmEngine(BaseEngine):
"disable_log_requests": True,
"enforce_eager": model_args.vllm_enforce_eager,
"enable_lora": model_args.adapter_name_or_path is not None,
"max_lora_rank": model_args.vllm_max_lora_rank,
}
if model_args.visual_inputs:

View File

@ -117,6 +117,7 @@ class ModelArguments:
default=False,
metadata={"help": "Whether or not to disable CUDA graph in the vLLM engine."},
)
vllm_max_lora_rank: int = field(default=8, metadata={"help": "The maximum supported rank of all LoRAs."})
offload_folder: str = field(
default="offload",
metadata={"help": "Path to offload model weights."},