diff --git a/paddlespeech/t2s/exps/syn_utils.py b/paddlespeech/t2s/exps/syn_utils.py index 751845912..4f2d1246d 100644 --- a/paddlespeech/t2s/exps/syn_utils.py +++ b/paddlespeech/t2s/exps/syn_utils.py @@ -592,9 +592,10 @@ def get_predictor( # after paddle 3.0, support new inference interface if paddlespeech.utils.satisfy_paddle_version('3.0.0-beta'): - model_name = model_file.split('.')[0] - assert model_name == params_file.split('.')[ - 0], "The prefix of model_file and params_file should be same." + model_name = str(model_file).rstrip('.pdmodel') + assert model_name == str(params_file).rstrip( + '.pdiparams' + ), "The prefix of model_file and params_file should be same." config = inference.Config(model_dir, model_name) else: config = inference.Config(