Skip to content

Commit df9d546

Browse files
committed
Fix edge case for vllm
Signed-off-by: Yi-Fu Wu <[email protected]>
1 parent f9730c8 commit df9d546

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/run_vlm_grpo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def hf_data_processor(
200200

201201
length = sum(len(m["token_ids"]) for m in message_log)
202202
loss_multiplier = 1.0
203-
if length > max_seq_length:
203+
if length >= max_seq_length:
204204
# Treat truncated messages as text only
205205
vllm_kwargs = {
206206
"vllm_content": None,

0 commit comments

Comments
 (0)