fix save inf (#52632)

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,