浏览代码

Update decoding.py (#1219)

Jong Wook Kim 2 年之前
父节点
当前提交
c09a7ae299
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 1 3
      whisper/decoding.py

+ 1 - 3
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: