diff --git a/src/llama-arch.cpp b/src/llama-arch.cpp index e1c5a27f7b..bdf2709175 100644 --- a/src/llama-arch.cpp +++ b/src/llama-arch.cpp @@ -2388,8 +2388,12 @@ LLM_TN_IMPL::LLM_TN_IMPL(llm_arch arch, llm_tensor tensor, const char * suffix, model_tensors(llm_get_tensor_names(arch)) {} std::string LLM_TN_IMPL::str() const { + if (LLM_TENSOR_NAMES.find(tensor) == LLM_TENSOR_NAMES.end()) { + GGML_ABORT("unknown tensor name for tensor id %d", static_cast(tensor)); + } + if (model_tensors.find(tensor) == model_tensors.end()) { - return "__missing__"; + return LLM_TENSOR_NAMES.at(tensor); } std::string name = ::format(LLM_TENSOR_NAMES.at(tensor), bid, xid);