chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

clamp end_timestamp to prevent infinite loop

Browse Source
This commit is contained in:
m-bain
2023-04-11 20:15:37 +01:00
committed by GitHub
parent 48ed89834e
commit 6a72b61564
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisperx/asr.py
View File

@ -269,6 +269,10 @@ def transcribe(
end_timestamp_pos = ( end_timestamp_pos = (
sliced_tokens[-1].item() - tokenizer.timestamp_begin sliced_tokens[-1].item() - tokenizer.timestamp_begin
) )
# clamp end-time to at least be 1 frame after start-time
end_timestamp_pos = max(end_timestamp_pos, start_timestamp_pos + time_precision)
current_segments.append( current_segments.append(
new_segment( new_segment(
start=time_offset + start_timestamp_pos * time_precision, start=time_offset + start_timestamp_pos * time_precision,