predict batch returns wrong results when given a batch of images

predict batch returns wrong results when given a batch of images: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). I'm trying to speed up inference by running prediction on a stack of images at once instead of looping over them in Python. I have a bunch of…

The task

I'm trying to speed up inference by running prediction on a stack of images at once instead of looping over them in Python. I have a bunch of pre-processed image tensors stacked into shape `(B, C, H, W)` and I'm calling:

Part of FineEnvs/MiMo-V2.6-RL-harbor-code.