fix bug for quantize minicpm

2024-02-06 00:41:52 +08:00 · 2024-02-06 00:41:52 +08:00 · b698d87e9a
commit b698d87e9a
parent 36084596c1
1 changed files with 1 additions and 0 deletions
--- a/llama.cpp
+++ b/llama.cpp
@ -697,6 +697,7 @@ static std::map<llm_arch, std::map<llm_tensor, std::string>> LLM_TENSOR_NAMES =
        {
            { LLM_TENSOR_TOKEN_EMBD,      "token_embd" },
            { LLM_TENSOR_OUTPUT_NORM,     "output_norm" },
+            { LLM_TENSOR_OUTPUT,          "output" },
            { LLM_TENSOR_ROPE_FREQS,      "rope_freqs" },
            { LLM_TENSOR_ATTN_NORM,       "blk.%d.attn_norm" },
            { LLM_TENSOR_ATTN_Q,          "blk.%d.attn_q" },