ReProver icon indicating copy to clipboard operation
ReProver copied to clipboard

Evaluation on multiple GPUs

Open ezhang7423 opened this issue 1 year ago • 1 comments

Hi there! I am trying to run evaluation with multiple GPUs. I have ran everything in './scripts/minimal_example.sh'. Running on 1 GPU works perfectly, however when I pass in more than 1 GPU, all Huggingface Generator workers still only initialize on the first GPU, leaving the rest unutilized. Unfortunately, --use-vlm does not appear to be working either, as an error is thrown that the T5 model is not supported. Any support would be greatly appreciated.

ezhang7423 avatar Sep 22 '24 20:09 ezhang7423