diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index ff55e968..a509c879 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -928,7 +928,7 @@ Status GeGenerator::Impl::SaveRootModel(const string &file_name_prefix, GeRootMo
     GELOGE(ret, "Save to om model failed");
     return ret;
   }
-  DisplayModelInfo(*(model_root.get()));
+  // DisplayModelInfo(*(model_root.get()));
   return SUCCESS;
 }
 
diff --git a/ge/offline/main.cc b/ge/offline/main.cc
index 74837e7a..55478991 100755
--- a/ge/offline/main.cc
+++ b/ge/offline/main.cc
@@ -1180,6 +1180,11 @@ domi::Status GenerateOmModel() {
     return domi::FAILED;
   }
 
+  if (FLAGS_display_model_info == "1") {
+    GELOGI("need to display model info.");
+    return ge::ConvertOm(FLAGS_output.c_str(), "", false);
+  }
+
   return domi::SUCCESS;
 }