From c8d2fef198e68329a33ae86948f1838443042017 Mon Sep 17 00:00:00 2001
From: huangjianhui <852142024@qq.com>
Date: Fri, 6 Jan 2023 13:48:44 +0800
Subject: [PATCH] [Bug Fix] When loading model from memory, don't need
 CheckModelFormat (#1073)

When loading model from memory, don't need CheckModelFormat
---
 fastdeploy/fastdeploy_model.cc | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/fastdeploy/fastdeploy_model.cc b/fastdeploy/fastdeploy_model.cc
index d0f06a1ca..05bc9b817 100755
--- a/fastdeploy/fastdeploy_model.cc
+++ b/fastdeploy/fastdeploy_model.cc
@@ -140,9 +140,11 @@ bool FastDeployModel::InitRuntimeWithSpecifiedDevice() {
 }
 
 bool FastDeployModel::InitRuntime() {
-  FDASSERT(
-      CheckModelFormat(runtime_option.model_file, runtime_option.model_format),
-      "ModelFormatCheck Failed.");
+  if (!runtime_option.model_from_memory_) {
+    FDASSERT(
+        CheckModelFormat(runtime_option.model_file, runtime_option.model_format),
+        "ModelFormatCheck Failed.");
+  }
   if (runtime_initialized_) {
     FDERROR << "The model is already initialized, cannot be initliazed again."
             << std::endl;