diff --git a/ge/common/helper/model_helper.cc b/ge/common/helper/model_helper.cc
index 7e54a224..abd1a4f5 100644
--- a/ge/common/helper/model_helper.cc
+++ b/ge/common/helper/model_helper.cc
@@ -79,9 +79,14 @@ Status ModelHelper::SaveModelPartition(std::shared_ptr<OmFileSaveHelper> &om_fil
 Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model) {
   vector<int64_t> om_info;
   ModelPtr model_tmp = ge::MakeShared<ge::Model>(ge_model->GetName(), ge_model->GetPlatformVersion());
+  if (model_tmp == nullptr) {
+    GELOGE(FAILED, "Create Model %s Ptr failed", ge_model->GetName().c_str());
+    return FAILED;
+  }
+  model_tmp->SetGraph(ge_model->GetGraph());
   model_tmp->SetVersion(ge_model->GetVersion());
   model_tmp->SetAttr(ge_model->MutableAttrMap());
-  ge::Buffer model_buffer
+  ge::Buffer model_buffer;
   (void)model_tmp->Save(model_buffer);
   GELOGD("SaveSizeToModelDef modeldef_size is %zu", model_buffer.GetSize());
   om_info.push_back(model_buffer.GetSize());