Browse Source

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

Vicki Anand 2 years ago
parent
commit
2b0c2971af
1 changed files with 1 additions and 1 deletions
  1. 1 1
      whisper/transcribe.py

+ 1 - 1
whisper/transcribe.py

@@ -220,7 +220,7 @@ def transcribe(
             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