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…