From 5ab7927325ea7e77d950a3cf38d9f7cd6d1a483e Mon Sep 17 00:00:00 2001
From: wangxiaoning <71813629+wangxn12138@users.noreply.github.com>
Date: Tue, 11 Apr 2023 15:13:13 +0800
Subject: [PATCH] fix save inf (#52632)

---
 .../distributed/fleet/runtime/parameter_server_runtime.py    | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/python/paddle/distributed/fleet/runtime/parameter_server_runtime.py b/python/paddle/distributed/fleet/runtime/parameter_server_runtime.py
index 24df3203183..37765833715 100644
--- a/python/paddle/distributed/fleet/runtime/parameter_server_runtime.py
+++ b/python/paddle/distributed/fleet/runtime/parameter_server_runtime.py
@@ -24,7 +24,6 @@ from paddle.static import (
     Variable,
     default_main_program,
     default_startup_program,
-    save_inference_model,
 )
 
 from ..base.private_helper_function import wait_server_ready
@@ -735,7 +734,7 @@ class ParameterServerRuntime(RuntimeBase):
                 raise TypeError(
                     "in fleet.save_inference_model() function, main_program must be as Program type, CompiledProgram is not allowed"
                 )
-            save_inference_model(
+            paddle.fluid.io.save_inference_model(
                 dirname,
                 feeded_var_names,
                 target_vars,
@@ -746,7 +745,7 @@ class ParameterServerRuntime(RuntimeBase):
                 export_for_deployment,
             )
         else:
-            save_inference_model(
+            paddle.fluid.io.save_inference_model(
                 dirname,
                 feeded_var_names,
                 target_vars,
-- 
GitLab