Ethan Rosenthal

Device mismatch during evaluation when training on mps

7

I ran into an issue trying to train flan-t5 on an M1 using torchmetrics. Training metrics worked fine, but I got the following stacktrace when calculating evaluation metrics: ```commandline ......