[Pipelines] Fix ASR model types check (#14178)

25ceb818 · Anton Lozhkov · GitHub · 6200fd7b · 25ceb818
Unverified Commit 25ceb818 authored Oct 27, 2021 by Anton Lozhkov Committed by GitHub Oct 27, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/pipelines/automatic_speech_recognition.py src/transformers/pipelines/automatic_speech_recognition.py +1 -1

No files found.
--- a/src/transformers/pipelines/automatic_speech_recognition.py
+++ b/src/transformers/pipelines/automatic_speech_recognition.py
@@ -106,7 +106,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
        if self.framework == "tf":
            raise ValueError("The AutomaticSpeechRecognitionPipeline is only available in PyTorch.")

-        self.check_model_type(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING.items() + MODEL_FOR_CTC_MAPPING.items())
+        self.check_model_type(dict(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING.items() + MODEL_FOR_CTC_MAPPING.items()))

    def __call__(
        self,