model to device in eval

qwen_classifier/evaluate.py

@@ -140,6 +140,7 @@ def _evaluate_local(test_data_path, hf_repo):
     with torch.no_grad():
         for batch in dataloader:
             print(f"EVALUATION RUNNING ON {global_model.device}")
+            global_model = global_model.to(DEVICE)
             batch = {k: v.to(DEVICE) for k, v in batch.items()}
             labels = batch["labels"].type(torch.float32)