diff --git a/source/api_cc/src/DeepPot.cc b/source/api_cc/src/DeepPot.cc index d86f63fe82..ba1f102084 100644 --- a/source/api_cc/src/DeepPot.cc +++ b/source/api_cc/src/DeepPot.cc @@ -1260,6 +1260,9 @@ void DeepPotModelDevi::init(const std::vector& 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]); }