[Fix] inference of paddle 3.0

pull/3963/head
megemini 8 months ago
parent 30d956d490
commit aaaf7d5686

@ -592,9 +592,10 @@ def get_predictor(
# after paddle 3.0, support new inference interface # after paddle 3.0, support new inference interface
if paddlespeech.utils.satisfy_paddle_version('3.0.0-beta'): if paddlespeech.utils.satisfy_paddle_version('3.0.0-beta'):
model_name = model_file.split('.')[0] model_name = str(model_file).rstrip('.pdmodel')
assert model_name == params_file.split('.')[ assert model_name == str(params_file).rstrip(
0], "The prefix of model_file and params_file should be same." '.pdiparams'
), "The prefix of model_file and params_file should be same."
config = inference.Config(model_dir, model_name) config = inference.Config(model_dir, model_name)
else: else:
config = inference.Config( config = inference.Config(

Loading…
Cancel
Save