Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

abort find_alignment on empty input

Browse Source
This commit is contained in:
Jong Wook Kim 2023-03-14 12:44:54 -07:00 committed by GitHub
parent 5f9ac653b7
commit 24ba319e70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats 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,