Merge branch 'master' into bugfix/default_uncollate

Lightning-Universe · Dec 15, 2021 · 48bdfe1 · 48bdfe1
2 parents 6aa2267 + 7bcf4f1
commit 48bdfe1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flash/audio/speech_recognition/collate.py b/flash/audio/speech_recognition/collate.py
@@ -20,9 +20,9 @@
 from flash.core.utilities.imports import _AUDIO_AVAILABLE
 
 if _AUDIO_AVAILABLE:
-    from transformers import Wav2Vec2Processor
+    from transformers import AutoProcessor
 else:
-    Wav2Vec2Processor = object
+    AutoProcessor = object
 
 
 @dataclass
@@ -52,7 +52,7 @@ class DataCollatorCTCWithPadding:
             7.5 (Volta).
     """
 
-    processor: Wav2Vec2Processor
+    processor: AutoProcessor
     padding: Union[bool, str] = True
     max_length: Optional[int] = None
     max_length_labels: Optional[int] = None