From 550ab5e1c5dc97208e7e148c8f81b1e06f0900b9 Mon Sep 17 00:00:00 2001 From: vincent Date: Mon, 5 Feb 2024 22:16:39 +0800 Subject: [PATCH] fix tab/space typo. --- gguf-py/gguf/constants.py | 2 +- llama.cpp | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/gguf-py/gguf/constants.py b/gguf-py/gguf/constants.py index 3b8da1dd8..bc552e6b3 100644 --- a/gguf-py/gguf/constants.py +++ b/gguf-py/gguf/constants.py @@ -157,7 +157,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = { MODEL_ARCH.CODESHELL: "codeshell", MODEL_ARCH.ORION: "orion", MODEL_ARCH.INTERNLM2: "internlm2", - MODEL_ARCH.MINICPM: "minicpm", + MODEL_ARCH.MINICPM: "minicpm", } TENSOR_NAMES: dict[MODEL_TENSOR, str] = { diff --git a/llama.cpp b/llama.cpp index 1a7c3af5c..cc50ea303 100644 --- a/llama.cpp +++ b/llama.cpp @@ -205,7 +205,7 @@ enum llm_arch { LLM_ARCH_CODESHELL, LLM_ARCH_ORION, LLM_ARCH_INTERNLM2, - LLM_ARCH_MINICPM, + LLM_ARCH_MINICPM, LLM_ARCH_UNKNOWN, }; @@ -229,7 +229,7 @@ static std::map LLM_ARCH_NAMES = { { LLM_ARCH_CODESHELL, "codeshell" }, { LLM_ARCH_ORION, "orion" }, { LLM_ARCH_INTERNLM2, "internlm2" }, - { LLM_ARCH_MINICPM, "minicpm" }, + { LLM_ARCH_MINICPM, "minicpm" }, }; enum llm_kv { @@ -692,7 +692,7 @@ static std::map> LLM_TENSOR_NAMES = { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" }, }, }, - { + { LLM_ARCH_MINICPM, { { LLM_TENSOR_TOKEN_EMBD, "token_embd" }, @@ -7114,7 +7114,7 @@ static struct ggml_cgraph * llama_build_graph( { result = llm.build_internlm2(); } break; - case LLM_ARCH_MINICPM: + case LLM_ARCH_MINICPM: { result = llm.build_minicpm(); } break;