提交

kongdeqiang

5 天以前 28ccfbfc51068a663a80764e14074df5edf2b5ba

 funasr/tokenizer/hf_tokenizer.py

@@ -1,16 +1,13 @@

try:
   from transformers import AutoTokenizer
except:
   # print("If you want to use hugging, please `pip install -U transformers`")
   pass

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)

   tokenizer = AutoTokenizer.from_pretrained(init_param_path)
	
   return tokenizer

    return tokenizer