add vllm config

Former-commit-id: 95365f0ce4f362bde7de8b679b54b548d7055bfb
This commit is contained in:
hiyouga
2024-11-10 21:28:18 +08:00
parent fcb6283a72
commit 1e6f96508a
34 changed files with 44 additions and 34 deletions

View File

@@ -46,7 +46,7 @@ class DataArguments:
metadata={"help": "Path to the folder containing the images or videos. Defaults to `dataset_dir`."},
)
cutoff_len: int = field(
default=1024,
default=2048,
metadata={"help": "The cutoff length of the tokenized inputs in the dataset."},
)
train_on_prompt: bool = field(