Use test case that previously failed

huggingface · Nov 11, 2024 · a5d0035 · a5d0035
1 parent 66df86a
commit a5d0035
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/onnxruntime/test_quantization.py b/tests/onnxruntime/test_quantization.py
@@ -55,8 +55,9 @@ class ORTQuantizerTest(unittest.TestCase):
         },
         "ort_model_with_onnx_model_in_subfolder": {
             "model_or_path": ORTModelForFeatureExtraction.from_pretrained(
-                "sentence-transformers-testing/stsb-bert-tiny-onnx",
+                "sentence-transformers/all-MiniLM-L6-v2",
                 subfolder="onnx",
+                file_name="model.onnx",
             )
         },
     }