resize file_contents

njzjz · Jan 11, 2024 · 8700607 · 8700607
1 parent a6732b0
commit 8700607
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/source/api_cc/src/DeepPot.cc b/source/api_cc/src/DeepPot.cc
@@ -1260,6 +1260,9 @@ void DeepPotModelDevi::init(const std::vector<std::string>& models,
     throw deepmd::deepmd_exception("no model is specified");
   }
   dps.resize(numb_models);
+  if (file_contents.size() != numb_models) {
+    file_contents.resize(numb_models);
+  }
   for (unsigned int ii = 0; ii < numb_models; ++ii) {
     dps[ii] = DeepPot(models[ii], gpu_rank, file_contents[ii]);
   }