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',
'vocab_file':
'punc_vocab.txt',
},
}
},
"ernie_linear_p3_wudao-punc-zh": {
'1.0': {
@ -543,10 +543,26 @@ text_dynamic_pretrained_models = {
'ckpt/model_state.pdparams',
'vocab_file':
'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 --------------
# ---------------------------------

Loading…
Cancel
Save