We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0c30117 commit c4a4b8bCopy full SHA for c4a4b8b
1 file changed
verl/trainer/data_loader.py
@@ -72,7 +72,9 @@ def create_dataloader(config: DataConfig, tokenizer: PreTrainedTokenizer, proces
72
prompt_key=config.prompt_key,
73
answer_key=config.answer_key,
74
image_key=config.image_key,
75
+ video_key=config.video_key,
76
image_dir=config.image_dir,
77
+ video_fps=config.video_fps,
78
max_prompt_length=config.max_prompt_length,
79
truncation="right",
80
format_prompt=config.format_prompt,
0 commit comments