Cast multimodal forward_kwargs to compute dtype for bf16/fp16 training#5073

akshan-main wants to merge 1 commit intohuggingface:mainhuggingface/trl:mainfrom

akshan-main:fix_grpo_vlm_pixel_dtypeakshan-main/trl:fix_grpo_vlm_pixel_dtypeCopy head branch name to clipboard