Fix DinoV2 crash when batch_size > 1#429

jonatanklosko merged 3 commits intoelixir-nx:mainelixir-nx/bumblebee:mainfrom

zacharydenton:mainzacharydenton/bumblebee:mainCopy head branch name to clipboard