add transformer to cli infer

3 years ago · 021311c76b
parent ab9bc3c31e
commit 021311c76b
1 changed files with 10 additions and 0 deletions
--- a/paddlespeech/cli/asr/infer.py
+++ b/paddlespeech/cli/asr/infer.py
@ -46,6 +46,16 @@ pretrained_models = {
        'conf/conformer.yaml',
        'ckpt_path':
        'exp/conformer/checkpoints/wenetspeech',
+    },
+    "transformer_zh_16k": {
+        'url':
+        'https://paddlespeech.bj.bcebos.com/s2t/aishell/asr1/transformer.model.tar.gz',
+        'md5':
+        '4e8b63800c71040b9390b150e2a5d4c4',
+        'cfg_path':
+        'conf/transformer.yaml',
+        'ckpt_path':
+        'exp/transformer/checkpoints/avg_20',
    }
 }