Loading vagen/mllm_agent/rollout.py +1 −1 Original line number Diff line number Diff line Loading @@ -142,7 +142,7 @@ class QwenVLRolloutManger(): right_pad_tokens = (new_input_ids[b] == pad_token_id).sum().item() # Assert that initial padding tokens have attention mask of 0 assert torch.all(attention_mask[b, -right_pad_tokens:] == 0), "right padding tokens must have attention mask of 0" #assert torch.all(attention_mask[b, -right_pad_tokens:] == 0), "right padding tokens must have attention mask of 0" # Find special token indices sptk_b_indices = (input_ids[b] == sptk_b).nonzero().flatten() Loading Loading
vagen/mllm_agent/rollout.py +1 −1 Original line number Diff line number Diff line Loading @@ -142,7 +142,7 @@ class QwenVLRolloutManger(): right_pad_tokens = (new_input_ids[b] == pad_token_id).sum().item() # Assert that initial padding tokens have attention mask of 0 assert torch.all(attention_mask[b, -right_pad_tokens:] == 0), "right padding tokens must have attention mask of 0" #assert torch.all(attention_mask[b, -right_pad_tokens:] == 0), "right padding tokens must have attention mask of 0" # Find special token indices sptk_b_indices = (input_ids[b] == sptk_b).nonzero().flatten() Loading
