diff --git a/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py b/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
index fcd42b6615415089c863f52ebacb316e0421cd94..5b80bdb95d8639bcf21fc62b988bec26c7db4b0a 100644
--- a/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
+++ b/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
@@ -144,16 +144,9 @@ class DistributedTranspiler(Fleet):
                                     executor, main_program, None, None,
                                     export_for_deployment)
         else:
-            io.save_inference_model(
-                dirname,
-                feeded_var_names,
-                target_vars,
-                executor,
-                self._origin_program,
-                None,
-                None,
-                export_for_deployment,
-                model_only=True)
+            io.save_inference_model(dirname, feeded_var_names, target_vars,
+                                    executor, self._origin_program, None, None,
+                                    export_for_deployment, True)
 
             model_basename = "__model__"
             model_filename = os.path.join(dirname, model_basename)