in src/open-r1-multimodal/src/open_r1/trainer/vllm_grpo_trainer.py [0:0]
def _get_per_token_logps( self, model, input_ids, attention_mask, pixel_values, image_grid_thw, logits_to_keep,