search

illuin-tech / vidore-benchmark

Vision Document Retrieval (ViDoRe): Benchmark. Evaluation code for the ColPali paper.
https://huggingface.co/vidore
MIT License
137 stars 15 forks source link

Multi-GPU Error #47

Closed SpaceLearner closed 1 month ago

SpaceLearner commented 1 month ago

When running model vidore/colpali-v1.2 on full evaluation, it will occur RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:6 and cuda:0!

tonywu71 commented 1 month ago

Hi @SpaceLearner, I'm afraid multi GPU evaluation on this repository is not part of our immediate roadmap. But I'd be happy to review a PR if you feel like implementing this feature 👋🏼