Skip to content

Commit c4a4b8b

Browse files
authored
[data] Fix valset loading for videos (#482)
1 parent 0c30117 commit c4a4b8b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

verl/trainer/data_loader.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,9 @@ def create_dataloader(config: DataConfig, tokenizer: PreTrainedTokenizer, proces
7272
prompt_key=config.prompt_key,
7373
answer_key=config.answer_key,
7474
image_key=config.image_key,
75+
video_key=config.video_key,
7576
image_dir=config.image_dir,
77+
video_fps=config.video_fps,
7678
max_prompt_length=config.max_prompt_length,
7779
truncation="right",
7880
format_prompt=config.format_prompt,

0 commit comments

Comments
 (0)