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

MiscellaneousStuff · Sep 29, 2022 · 2b0c297 · 2b0c297
1 parent 62fe7f1
commit 2b0c297
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/whisper/transcribe.py b/whisper/transcribe.py
@@ -220,7 +220,7 @@ def add_segment(
  else:
  duration = segment_duration
  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.
  # single timestamp at the end means no speech after the last timestamp.
  last_timestamp_position = timestamps[-1].item() - tokenizer.timestamp_begin