From 24ba319e702fa37ff089ae29d9d453b64e9beffd Mon Sep 17 00:00:00 2001 From: Jong Wook Kim Date: Tue, 14 Mar 2023 12:44:54 -0700 Subject: [PATCH] abort find_alignment on empty input --- whisper/timing.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/whisper/timing.py b/whisper/timing.py index 1f8f4cf..f5d07e5 100644 --- a/whisper/timing.py +++ b/whisper/timing.py @@ -170,6 +170,9 @@ def find_alignment( medfilt_width: int = 7, qk_scale: float = 1.0, ) -> List[WordTiming]: + if len(text_tokens) == 0: + return [] + tokens = torch.tensor( [ *tokenizer.sot_sequence,