TypeError: default_collate: batch must contain tensors... found <class 'NoneType'> when running MTEB with vidore/colqwen2.5-v0.2

[ghazalkhalighinejad]

Running mteb run -b "ViDoRe(v2)" -m "vidore/colqwen2.5-v0.2" crashes with:

TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'NoneType'>

[QuentinJGMace]

Hey @ghazalkhalighinejad , will look into that, just in order to have more infos, on what device are you running this code / what are your transformers/torch versions ?

[ghazalkhalighinejad]

Hi @QuentinJGMace, thanks!

Device: NVIDIA A6000
Python: 3.12.8
Transformers: 4.53.3
Torch: 2.5.1+cu121