speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
语帆 2024-02-22 16:08:51 +08:00
parent 7f0a06946f
commit 733073d269
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/auto/auto_model.py
View File

@ -153,15 +153,18 @@ class AutoModel:
# build tokenizer # build tokenizer
tokenizer = kwargs.get("tokenizer", None) tokenizer = kwargs.get("tokenizer", None)
pdb.set_trace()
if tokenizer is not None: if tokenizer is not None:
tokenizer_class = tables.tokenizer_classes.get(tokenizer) tokenizer_class = tables.tokenizer_classes.get(tokenizer)
pdb.set_trace()
tokenizer = tokenizer_class(**kwargs["tokenizer_conf"]) tokenizer = tokenizer_class(**kwargs["tokenizer_conf"])
pdb.set_trace()
kwargs["tokenizer"] = tokenizer kwargs["tokenizer"] = tokenizer
kwargs["token_list"] = tokenizer.token_list kwargs["token_list"] = tokenizer.token_list
vocab_size = len(tokenizer.token_list) vocab_size = len(tokenizer.token_list)
else: else:
vocab_size = -1 vocab_size = -1
pdb.set_trace()
# build frontend # build frontend
frontend = kwargs.get("frontend", None) frontend = kwargs.get("frontend", None)
if frontend is not None: if frontend is not None: