abort find_alignment on empty input (#1090)

2023-03-14 15:47:58 -04:00
parent 5f9ac653b7
commit 79c43e4859
1 changed files with 3 additions and 0 deletions
--- 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,