Ajinkya Tejankar

Hi, I am interested in reproducing the results for VisDa-17 benchmark. I couldn't find any instructions on how to do this in the repository.

Fix eval_imagenet and add eval_winoground

6

1. Replace `.cuda()` calls with `.to()` calls and correct device in `eval_imagenet` function 2. Add `eval_winoground` to evaluate on the Winoground dataset

# (WIP) Fix for the LM_HEAD issue **Root Cause**. The error is caused by incorrect segments passed to the `lora_b_sgmv` kernel during the prefill stage. This happens because we do...