FunASR-bmodel / funasr /tokenizer /hf_tokenizer.py
mrfakename's picture
Mirror wlc952/FunASR-bmodel from ModelScope (part 2)
c14c4fe verified
Raw History Blame Contribute Delete
402 Bytes
from funasr.register import tables
@tables.register("tokenizer_classes", "HuggingfaceTokenizer")
def HuggingfaceTokenizer(init_param_path, **kwargs):
try:
from transformers import AutoTokenizer
except:
# print("If you want to use hugging, please `pip install -U transformers`")
pass
tokenizer = AutoTokenizer.from_pretrained(init_param_path)
return tokenizer