diff --git a/ge/session/omg.cc b/ge/session/omg.cc
index c9da9a40..35939553 100755
--- a/ge/session/omg.cc
+++ b/ge/session/omg.cc
@@ -910,6 +910,12 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
 
   // om info
   iter = model_attr_map->find("om_info_list");
+  if (iter == model_attr_map->end()) {
+    std::cout << "Display Model Info error, om_info_list is not find in om, please check atc version is matched."
+              << std::endl;
+    std::cout << "------------ Display Model Info end   ------------"  << std::endl;
+    return;
+  }
   auto bt = iter->second.list().i_size();
   if (bt == 5) {
     std::cout << "om info: "