Add Speech AutoModels (#13655)

* upload * correct * correct * correct * finish * up * up * up again
2021-09-21 08:50:33 +02:00
parent ea92136597
commit 48fa42e5d5
8 changed files with 91 additions and 8 deletions
--- a/tests/test_pipelines_automatic_speech_recognition.py
+++ b/tests/test_pipelines_automatic_speech_recognition.py
@@ -49,10 +49,10 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase):
    @require_torch
    def test_torch_small_no_tokenizer_files(self):
        # test that model without tokenizer file cannot be loaded
-        with pytest.raises(ValueError):
+        with pytest.raises(OSError):
            pipeline(
                task="automatic-speech-recognition",
-                model="hf-internal-testing/tiny-random-wav2vec2",
+                model="patrickvonplaten/tiny-wav2vec2-no-tokenizer",
                framework="pt",
            )