mirror of
https://gitee.com/270580156/weiyu.git
synced 2026-05-20 14:18:01 +00:00
Sync from bytedesk-private: update
This commit is contained in:
13
modules/python/vendors/FunASR/funasr/tokenizer/hf_tokenizer.py
vendored
Normal file
13
modules/python/vendors/FunASR/funasr/tokenizer/hf_tokenizer.py
vendored
Normal file
@@ -0,0 +1,13 @@
|
||||
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
|
||||
Reference in New Issue
Block a user