Spaces:
Sleeping
Sleeping
Update mmlu_pro_eval_adapted.py
Browse files- mmlu_pro_eval_adapted.py +0 -3
mmlu_pro_eval_adapted.py
CHANGED
@@ -199,9 +199,6 @@ def evaluate_mmlu_pro(model_name, num_subjects=-1, num_questions=10, num_shots=5
|
|
199 |
|
200 |
model, tokenizer = load_model(model_name, gpu_utilization=0.8)
|
201 |
|
202 |
-
# Ensure model is in evaluation mode
|
203 |
-
model[0].model.eval() # Assuming model is a tuple of (llm, sampling_params)
|
204 |
-
|
205 |
test_df, val_df = load_mmlu_pro()
|
206 |
|
207 |
test_df = pd.DataFrame(test_df)
|
|
|
199 |
|
200 |
model, tokenizer = load_model(model_name, gpu_utilization=0.8)
|
201 |
|
|
|
|
|
|
|
202 |
test_df, val_df = load_mmlu_pro()
|
203 |
|
204 |
test_df = pd.DataFrame(test_df)
|