Update pretrained_models.py

Add a new model for faster text process
pull/2421/head
Zhao Yuting 2 years ago committed by GitHub
parent a17baad534
commit a02654660a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -529,7 +529,7 @@ text_dynamic_pretrained_models = {
'ckpt/model_state.pdparams', 'ckpt/model_state.pdparams',
'vocab_file': 'vocab_file':
'punc_vocab.txt', 'punc_vocab.txt',
}, }
}, },
"ernie_linear_p3_wudao-punc-zh": { "ernie_linear_p3_wudao-punc-zh": {
'1.0': { '1.0': {
@ -543,10 +543,26 @@ text_dynamic_pretrained_models = {
'ckpt/model_state.pdparams', 'ckpt/model_state.pdparams',
'vocab_file': 'vocab_file':
'punc_vocab.txt', 'punc_vocab.txt',
}
}, },
}, "ernie_linear_p3_wudao_fast-punc-zh": {
'1.0':{
'url':
'https://paddlespeech.bj.bcebos.com/text/ernie_linear_p3_wudao_fast-punc-zh.tar.gz',
'md5':
'c93f9594119541a5dbd763381a751d08',
'cfg_path':
'ckpt/model_config.json',
'ckpt_path':
'ckpt/model_state.pdparams',
'vocab_file':
'punc_vocab.txt',
}
}
} }
# --------------------------------- # ---------------------------------
# -------------- TTS -------------- # -------------- TTS --------------
# --------------------------------- # ---------------------------------

Loading…
Cancel
Save