Skip to content

GitLab

Unverified Commit 25ceb818 authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

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

parent 6200fd7b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/pipelines/automatic_speech_recognition.py
View file @ 25ceb818
...@@ -106,7 +106,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline): ...@@ -106,7 +106,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
if self.framework == "tf": if self.framework == "tf":
raise ValueError("The AutomaticSpeechRecognitionPipeline is only available in PyTorch.") 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__( def __call__(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment