From 1a807fc44ce607e61160b2f4b2e5c2573925dc92 Mon Sep 17 00:00:00 2001 From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> Date: Mon, 18 Dec 2023 11:37:24 +0100 Subject: [PATCH] Modify SEW-D model for ORT tests (#1601) --- tests/onnxruntime/utils_onnxruntime_tests.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/onnxruntime/utils_onnxruntime_tests.py b/tests/onnxruntime/utils_onnxruntime_tests.py index 8e579879ea6..c71a3243674 100644 --- a/tests/onnxruntime/utils_onnxruntime_tests.py +++ b/tests/onnxruntime/utils_onnxruntime_tests.py @@ -95,7 +95,7 @@ "roformer": "hf-internal-testing/tiny-random-RoFormerModel", "segformer": "hf-internal-testing/tiny-random-SegformerModel", "sew": "hf-internal-testing/tiny-random-SEWModel", - "sew_d": "hf-internal-testing/tiny-random-SEWDModel", + "sew_d": "asapp/sew-d-tiny-100k-ft-ls100h", "squeezebert": "hf-internal-testing/tiny-random-SqueezeBertModel", "speech_to_text": "hf-internal-testing/tiny-random-Speech2TextModel", "stable-diffusion": "hf-internal-testing/tiny-stable-diffusion-torch",