Update mmlu_pro_eval_adapted.py

mmlu_pro_eval_adapted.py

@@ -258,7 +258,8 @@ def evaluate_mmlu_pro(model_name, num_subjects=-1, num_questions=10, num_shots=5
         num_questions: Number of questions per subject
         num_shots: Number of examples to include in prompts
     """
-    print("IS CUDA AVAILABLE: ", torch.cuda.is_available())
+    print(f"Is CUDA available: {torch.cuda.is_available()}")
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
     
     # Load model and data
     model, tokenizer = load_model(model_name, gpu_utilization=0.8)