chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

suppress numerals change logic

Browse Source
This commit is contained in:
Max Bain
2023-06-05 15:44:17 +01:00
parent 74a00eecd7
commit d7f1d16f19
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisperx/asr.py
View File

@ -17,7 +17,8 @@ def find_numeral_symbol_tokens(tokenizer):
numeral_symbol_tokens = [] numeral_symbol_tokens = []
for i in range(tokenizer.eot): for i in range(tokenizer.eot):
token = tokenizer.decode([i]).removeprefix(" ") token = tokenizer.decode([i]).removeprefix(" ")
if all(c in "0123456789@#%&*+=_$:-.,?!" for c in token): has_numeral_symbol = any(c in "0123456789%$£" for c in token)
if has_numeral_symbol:
numeral_symbol_tokens.append(i) numeral_symbol_tokens.append(i)
return numeral_symbol_tokens return numeral_symbol_tokens