pull/4101/head
zxcd 1 week ago
parent 38b3c4a87b
commit 40b2bd109f

@ -700,6 +700,7 @@ def transcribe(
consecutive = paddle.where(timestamp_tokens[:-1] & timestamp_tokens[ consecutive = paddle.where(timestamp_tokens[:-1] & timestamp_tokens[
1:])[0] 1:])[0]
if consecutive.numel() != 0:
consecutive = paddle.add(consecutive, paddle.to_tensor(1)) consecutive = paddle.add(consecutive, paddle.to_tensor(1))
if len(consecutive) > 0: if len(consecutive) > 0:
# if the output contains two consecutive timestamp tokens # if the output contains two consecutive timestamp tokens

Loading…
Cancel
Save