Merge pull request #1109 from yt605155624/tts_cli

[cli]update voc name
3 years ago · 3fe75f833d
parent e98e80e82a ca12a83d5a
commit 3fe75f833d
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/cli/tts/infer.py
+++ b/paddlespeech/cli/tts/infer.py
@ -467,7 +467,7 @@ class TTSExecutor(BaseExecutor):

        # vocoder
        # model: {model_name}_{dataset}
-        voc_name = '_'.join(voc.split('_')[:-1])
+        voc_name = voc[:voc.rindex('_')]
        voc_class = dynamic_import(voc_name, model_alias)
        voc_inference_class = dynamic_import(voc_name + '_inference',
                                             model_alias)