wav2vec2 backend + padding.

brave-experiments · Aug 19, 2024 · 88c27d7 · 88c27d7
1 parent 343cae5
commit 88c27d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/runners/audio_transcriber.py b/src/runners/audio_transcriber.py
@@ -187,7 +187,7 @@ def __init__(self):
 
     def transcribe(self, audios):
         input_values = self.processor(
-            audios, return_tensors="pt", sampling_rate=16000
+            audios, return_tensors="pt", sampling_rate=16000, padding=True
         ).input_values.cuda()
 
         with torch.no_grad():