add vllm config

This commit is contained in:
hiyouga
2024-11-10 21:28:18 +08:00
parent adc5849ce7
commit 58ab4579dc
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(