Update decoding.py (#1219)

2023-04-11 18:13:13 -04:00
parent b0022b3283
commit c09a7ae299
1 changed files with 1 additions and 3 deletions
--- a/whisper/decoding.py
+++ b/whisper/decoding.py
@@ -469,9 +469,7 @@ class ApplyTimestampRules(LogitFilter):
            ]
            if timestamps.numel() > 0:
                # timestamps shouldn't decrease; forbid timestamp tokens smaller than the last
-                logits[k, self.tokenizer.timestamp_begin : timestamps[-1]] = -np.inf
-
-                # to force that timestamps are strictly increasing
+                # also force each segment to have a nonzero length, to prevent infinite looping
                if last_was_timestamp and not penultimate_was_timestamp:
                    timestamp_last = timestamps[-1]
                else: