gaspersic/whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't update duration if last timestamp is same as begin (#191)

Browse Source
This commit is contained in:
Vicki Anand
2022-09-29 15:27:48 -04:00
committed by GitHub
parent 62fe7f1009
commit 2b0c2971af
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/transcribe.py
View File

@@ -220,7 +220,7 @@ def transcribe(
else: else:
duration = segment_duration duration = segment_duration
timestamps = tokens[timestamp_tokens.nonzero().flatten()] timestamps = tokens[timestamp_tokens.nonzero().flatten()]
if len(timestamps) > 0: if len(timestamps) > 0 and timestamps[-1].item() != tokenizer.timestamp_begin:
# no consecutive timestamps but it has a timestamp; use the last one. # no consecutive timestamps but it has a timestamp; use the last one.
# single timestamp at the end means no speech after the last timestamp. # single timestamp at the end means no speech after the last timestamp.
last_timestamp_position = timestamps[-1].item() - tokenizer.timestamp_begin last_timestamp_position = timestamps[-1].item() - tokenizer.timestamp_begin