Merge pull request #3119 from alibaba/feature/bugfix

MNN:Bugfix: Fix bug for Quantize Static model load error
alibaba · Dec 10, 2024 · 4c75ee7 · 4c75ee7
2 parents 2b899c1 + cba6e25
commit 4c75ee7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/source/utils/InitNet.cpp b/source/utils/InitNet.cpp
@@ -204,6 +204,9 @@ bool initTensors(std::vector<std::shared_ptr<Tensor>>& tensors, const Net* net,
         }
         auto blob = des->blob();
         auto& tb = tensors[index]->buffer();
+        if (nullptr == blob) {
+            continue;
+        }
         if (auto idims = blob->dims()) {
             for (int d = 0; d < idims->size(); d++) {
                 tb.dim[d].extent = idims->Get(d);