We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f7e47d6 commit 731da68Copy full SHA for 731da68
1 file changed
verl/utils/rl_dataset.py
@@ -120,7 +120,9 @@ def __getitem__(self, index):
120
index = 0
121
while "<image>" in prompt:
122
prompt = prompt.replace(
123
- "<image>", "<|placeholder|>" * (image_grid_thw[index].prod() // merge_length), 1
+ "<image>",
124
+ "<|vision_start|>" + "<|placeholder|>" * (image_grid_thw[index].prod() // merge_length) + "<|vision_end|>",
125
+ 1
126
)
127
index += 1
128
0 commit comments