MNN:Bugfix: Fix bug for alibaba#3058

mcx · Dec 10, 2024 · cba6e25 · cba6e25
1 parent 2b899c1
commit cba6e25
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);