Update model_args.py

2024-05-31 14:35:48 +08:00 · 2024-05-31 14:35:48 +08:00 · dade2f083d
parent c4f50865ad
commit dade2f083d
1 changed files with 1 additions and 1 deletions
--- a/src/llamafactory/hparams/model_args.py
+++ b/src/llamafactory/hparams/model_args.py
@ -107,7 +107,7 @@ class ModelArguments:
    )
    vllm_maxlen: int = field(
        default=2048,
-        metadata={"help": "Maximum input length of the vLLM engine."},
+        metadata={"help": "Maximum sequence length of the vLLM engine (including prompt and output)."},
    )
    vllm_gpu_util: float = field(
        default=0.9,