Update decoding.py (openai#1219)

wlrnet · Apr 11, 2023 · c09a7ae · c09a7ae
1 parent b0022b3
commit c09a7ae
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/whisper/decoding.py b/whisper/decoding.py
@@ -469,9 +469,7 @@ def apply(self, logits: Tensor, tokens: Tensor):
             ]
             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: