fix

huggingface · Jul 4, 2024 · 4d8a766 · 4d8a766
1 parent 9699f23
commit 4d8a766
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/examples/onnxruntime/optimization/multiple-choice/run_swag.py b/examples/onnxruntime/optimization/multiple-choice/run_swag.py
@@ -346,7 +346,7 @@ def compute_metrics(eval_predictions):
         outputs = evaluation_loop(
             model=model,
             dataset=eval_dataset,
-            label_names=["labels"],
+            label_names=["label"],
             compute_metrics=compute_metrics,
         )
 

diff --git a/examples/onnxruntime/quantization/question-answering/run_qa.py b/examples/onnxruntime/quantization/question-answering/run_qa.py
@@ -683,7 +683,6 @@ def compute_metrics(p: EvalPrediction):
         outputs = evaluation_loop(
             model=model,
             dataset=predict_dataset,
-            compute_metrics=compute_metrics,
             label_names=["start_positions", "end_positions"],
         )
         predictions = post_processing_function(predict_examples, predict_dataset, outputs.predictions)