Update logic get segment from features before encoding

SYSTRAN · Feb 23, 2024 · e20cc93 · e20cc93
1 parent 06d32bf
commit e20cc93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
@@ -490,7 +490,7 @@ def generate_segments(
                 content_frames - seek,
                 seek_clip_end - seek,
             )
-            segment = features[:, seek : seek + segment_size]
+            segment = features[:, seek: seek + self.feature_extractor.nb_max_frames]
             segment_duration = segment_size * self.feature_extractor.time_per_frame
 
             if self.logger.isEnabledFor(logging.DEBUG):