update voc name

3 years ago · ca12a83d5a
parent 2189b46004
commit ca12a83d5a
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/cli/tts/infer.py
+++ b/paddlespeech/cli/tts/infer.py
@ -465,7 +465,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)