From 426daca661fa5799cb3164ae9def95d578f5e423 Mon Sep 17 00:00:00 2001 From: trungkienbkhn Date: Fri, 23 Feb 2024 17:29:13 +0700 Subject: [PATCH] Update logic get segment from features before encoding --- faster_whisper/transcribe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py index c1ea390b..f82e79b6 100644 --- 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):