From e4a2accf4ab569ce3a41860f7bcb972e14e7ec10 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Fri, 17 May 2024 16:16:41 +0800 Subject: [PATCH] Update model_args.py Former-commit-id: 964672745389e35580a7010b0f010bd5ee08d542 --- src/llamafactory/hparams/model_args.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/llamafactory/hparams/model_args.py b/src/llamafactory/hparams/model_args.py index 20ff74aa..5885bb09 100644 --- a/src/llamafactory/hparams/model_args.py +++ b/src/llamafactory/hparams/model_args.py @@ -117,7 +117,10 @@ 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."}) + vllm_max_lora_rank: int = field( + default=8, + metadata={"help": "Maximum rank of all LoRAs in the vLLM engine."}, + ) offload_folder: str = field( default="offload", metadata={"help": "Path to offload model weights."},