From ca38cff71dd0b4e085b000664e2428a75ad68aec Mon Sep 17 00:00:00 2001
From: xutianchun <xutianchun@huawei.com>
Date: Thu, 25 Mar 2021 09:52:58 +0800
Subject: [PATCH] fix model buf may double free

---
 mindspore/lite/src/lite_model.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/lite/src/lite_model.cc b/mindspore/lite/src/lite_model.cc
index 5863459720..ea4195af8f 100644
--- a/mindspore/lite/src/lite_model.cc
+++ b/mindspore/lite/src/lite_model.cc
@@ -333,6 +333,9 @@ Model *ImportFromBuffer(const char *model_buf, size_t size, bool take_buf) {
   model->buf_size_ = size;
   auto status = model->ConstructModel();
   if (status != RET_OK) {
+    if (take_buf) {
+      model->buf = nullptr;
+    }
     MS_LOG(ERROR) << "construct model failed.";
     delete model;
     return nullptr;