Update mmlu_pro_eval_adapted.py

mmlu_pro_eval_adapted.py

@@ -18,7 +18,6 @@ import pandas as pd
 import numpy as np
 
 logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
 
 # Can be found at https://github.com/TIGER-AI-Lab/MMLU-Pro/blob/main/cot_prompt_lib/initial_prompt.txt
 initial_prompt = "The following are multiple choice questions (with answers) about {$}. Think step by step and then finish your answer with \"the answer is (X)\" where X is the correct letter choice."
@@ -59,7 +58,7 @@ def load_model(model_name, gpu_utilization=0.8):
                 tensor_parallel_size=torch.cuda.device_count(),
                 max_model_len=max_model_length,
                 trust_remote_code=True)
-    logger.info(f"Torch Device CUDA Count: {torch.cuda.device_count()}")
+    logging.info(f"Torch Device CUDA Count: {torch.cuda.device_count()}")
     sampling_params = SamplingParams(temperature=0, max_tokens=max_new_tokens,
                                         stop=["Question:"])
     tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)