Skip to content

Commit 731da68

Browse files
authored
[data] fix vision tokens (#31)
Fix template issue
1 parent f7e47d6 commit 731da68

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

verl/utils/rl_dataset.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,9 @@ def __getitem__(self, index):
120120
index = 0
121121
while "<image>" in prompt:
122122
prompt = prompt.replace(
123-
"<image>", "<|placeholder|>" * (image_grid_thw[index].prod() // merge_length), 1
123+
"<image>",
124+
"<|vision_start|>" + "<|placeholder|>" * (image_grid_thw[index].prod() // merge_length) + "<|vision_end|>",
125+
1
124126
)
125127
index += 1
126128

0 commit comments

Comments
 (0)