heimoshuiyu/whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

abort find_alignment on empty input (#1090)

Browse Source
This commit is contained in:
Jong Wook Kim
2023-03-14 15:47:58 -04:00
committed by GitHub
parent 5f9ac653b7
commit 79c43e4859
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisper/timing.py
View File

@@ -170,6 +170,9 @@ def find_alignment(
medfilt_width: int = 7, medfilt_width: int = 7,
qk_scale: float = 1.0, qk_scale: float = 1.0,
) -> List[WordTiming]: ) -> List[WordTiming]:
if len(text_tokens) == 0:
return []
tokens = torch.tensor( tokens = torch.tensor(
[ [
*tokenizer.sot_sequence, *tokenizer.sot_sequence,