hf_tokenizer.py 363 B

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