雾聪
2024-03-05 6dcd960fda8be389af355ede4ecc583b036029d4
1
2
3
4
5
6
7
8
9
10
11
12
13
14
 
try:
    from transformers import AutoTokenizer
except:
    print("If you want to use hugging, please `pip install -U transformers`")
 
from funasr.register import tables
 
@tables.register("tokenizer_classes", "HuggingfaceTokenizer")
def HuggingfaceTokenizer(init_param_path, **kwargs):
 
    tokenizer = AutoTokenizer.from_pretrained(init_param_path)
    
    return tokenizer