abort find_alignment on empty input (openai#1090)

Hilokal · Mar 14, 2023 · 79c43e4 · 79c43e4
1 parent 5f9ac65
commit 79c43e4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,