..

SYSTRAN · Dec 12, 2024 · 4863a1f · 4863a1f
1 parent ce6e298
commit 4863a1f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/faster_whisper/vad.py b/faster_whisper/vad.py
@@ -302,13 +302,12 @@ def __call__(
 
         batched_audio = batched_audio.reshape(-1, num_samples + context_size_samples)
 
-        batch_process_size = 10000
+        encoder_batch_size = 10000
         num_segments = batched_audio.shape[0]
         encoder_outputs = []
-        for start in range(0, num_segments, batch_process_size):
-            end = min(start + batch_process_size, num_segments)
+        for i in range(0, num_segments, encoder_batch_size):
             encoder_output = self.encoder_session.run(
-                None, {"input": batched_audio[start:end]}
+                None, {"input": batched_audio[i : i + encoder_batch_size]}
             )[0]
             encoder_outputs.append(encoder_output)