udpate model id

huggingface · Sep 30, 2024 · e146328 · e146328
1 parent e7d3ba4
commit e146328
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -3147,7 +3147,7 @@ class ORTModelForAudioClassificationIntegrationTest(ORTModelTestMixin):
  "wavlm",
  "wav2vec2",
  "wav2vec2-conformer",
- # "whisper", saving of the model's config is broken in transformers v4.45.0
+ "whisper",
  ]
 
  FULL_GRID = {"model_arch": SUPPORTED_ARCHITECTURES}

diff --git a/tests/onnxruntime/utils_onnxruntime_tests.py b/tests/onnxruntime/utils_onnxruntime_tests.py
@@ -151,7 +151,7 @@
  "unispeech_sat": "hf-internal-testing/tiny-random-UnispeechSatModel",
  "vision-encoder-decoder": "hf-internal-testing/tiny-random-VisionEncoderDecoderModel-vit-gpt2",
  "vit": "hf-internal-testing/tiny-random-vit",
- "whisper": "openai/whisper-tiny.en", # hf-internal-testing ones are broken
+ "whisper": "optimum-internal-testing/tiny-random-whisper",
  "wav2vec2": "hf-internal-testing/tiny-random-Wav2Vec2Model",
  "wav2vec2-conformer": "hf-internal-testing/tiny-random-wav2vec2-conformer",
  "wavlm": "hf-internal-testing/tiny-random-WavlmModel",