diff --git a/PaddleNLP/PaddleMT/transformer/inference_model.py b/PaddleNLP/PaddleMT/transformer/inference_model.py index 5de0a107cd941c7d2eba42d1ec9922095bde5d8d..9ca71777e2b90ef03215527a50998b87d81f6cbc 100644 --- a/PaddleNLP/PaddleMT/transformer/inference_model.py +++ b/PaddleNLP/PaddleMT/transformer/inference_model.py @@ -131,7 +131,7 @@ def do_save_inference_model(args): fluid.io.save_inference_model( args.inference_model_dir, - feeded_var_names=input_field_names, + feeded_var_names=list(input_field_names), target_vars=[out_ids, out_scores], executor=exe, main_program=test_prog,