Skip to content

Commit 20c7e1e

Browse files
authored
gguf : fix a few general keys (#3341)
1 parent dc68974 commit 20c7e1e

File tree

4 files changed

+15
-15
lines changed

4 files changed

+15
-15
lines changed

examples/gptneox-wip/falcon-main.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -367,10 +367,10 @@ bool falcon_model_load(const std::string & fname, falcon_model & model, gpt2bpe_
367367
keyidx = gguf_find_key(ggufctx, "general.architecture");
368368
if (keyidx != -1) { printf("%s: model architecture = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
369369
keyidx = gguf_find_key(ggufctx, "general.file_type");
370-
if (keyidx != -1) { printf("%s: model file type = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
370+
if (keyidx != -1) { printf("%s: model file type = %" PRIu32 "\n", __func__, gguf_get_val_u32(ggufctx, keyidx)); }
371371
keyidx = gguf_find_key(ggufctx, "gptneox.tensor_data_layout");
372372
if (keyidx != -1) { printf("%s: model data layout = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
373-
keyidx = gguf_find_key(ggufctx, "general.source.hugginface.repository");
373+
keyidx = gguf_find_key(ggufctx, "general.source.huggingface.repository");
374374
if (keyidx != -1) { printf("%s: model source HF repo = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
375375
}
376376

examples/gptneox-wip/gptneox-main.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -380,10 +380,10 @@ bool gpt_neox_model_load(const std::string & fname, gpt_neox_model & model, gpt2
380380
keyidx = gguf_find_key(ggufctx, "general.architecture");
381381
if (keyidx != -1) { printf("%s: model architecture = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
382382
keyidx = gguf_find_key(ggufctx, "general.file_type");
383-
if (keyidx != -1) { printf("%s: model file type = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
383+
if (keyidx != -1) { printf("%s: model file type = %" PRIu32 "\n", __func__, gguf_get_val_u32(ggufctx, keyidx)); }
384384
keyidx = gguf_find_key(ggufctx, "gptneox.tensor_data_layout");
385385
if (keyidx != -1) { printf("%s: model data layout = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
386-
keyidx = gguf_find_key(ggufctx, "general.source.hugginface.repository");
386+
keyidx = gguf_find_key(ggufctx, "general.source.huggingface.repository");
387387
if (keyidx != -1) { printf("%s: model source HF repo = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); }
388388
}
389389

gguf-py/gguf/gguf.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
KEY_GENERAL_DESCRIPTION = "general.description"
3333
KEY_GENERAL_LICENSE = "general.license"
3434
KEY_GENERAL_SOURCE_URL = "general.source.url"
35-
KEY_GENERAL_SOURCE_HF_REPO = "general.source.hugginface.repository"
35+
KEY_GENERAL_SOURCE_HF_REPO = "general.source.huggingface.repository"
3636
KEY_GENERAL_FILE_TYPE = "general.file_type"
3737

3838
# LLM

llama.cpp

+10-10
Original file line numberDiff line numberDiff line change
@@ -221,16 +221,16 @@ enum llm_kv {
221221
};
222222

223223
static std::map<llm_kv, std::string> LLM_KV_NAMES = {
224-
{ LLM_KV_GENERAL_ARCHITECTURE, "general.architecture" },
225-
{ LLM_KV_GENERAL_QUANTIZATION_VERSION, "general.quantization_version" },
226-
{ LLM_KV_GENERAL_ALIGNMENT, "general.alignment" },
227-
{ LLM_KV_GENERAL_NAME, "general.name" },
228-
{ LLM_KV_GENERAL_AUTHOR, "general.author" },
229-
{ LLM_KV_GENERAL_URL, "general.url" },
230-
{ LLM_KV_GENERAL_DESCRIPTION, "general.description" },
231-
{ LLM_KV_GENERAL_LICENSE, "general.license" },
232-
{ LLM_KV_GENERAL_SOURCE_URL, "general.source_url" },
233-
{ LLM_KV_GENERAL_SOURCE_HF_REPO, "general.source_hf_repo" },
224+
{ LLM_KV_GENERAL_ARCHITECTURE, "general.architecture" },
225+
{ LLM_KV_GENERAL_QUANTIZATION_VERSION, "general.quantization_version" },
226+
{ LLM_KV_GENERAL_ALIGNMENT, "general.alignment" },
227+
{ LLM_KV_GENERAL_NAME, "general.name" },
228+
{ LLM_KV_GENERAL_AUTHOR, "general.author" },
229+
{ LLM_KV_GENERAL_URL, "general.url" },
230+
{ LLM_KV_GENERAL_DESCRIPTION, "general.description" },
231+
{ LLM_KV_GENERAL_LICENSE, "general.license" },
232+
{ LLM_KV_GENERAL_SOURCE_URL, "general.source.url" },
233+
{ LLM_KV_GENERAL_SOURCE_HF_REPO, "general.source.huggingface.repository" },
234234

235235
{ LLM_KV_CONTEXT_LENGTH, "%s.context_length" },
236236
{ LLM_KV_EMBEDDING_LENGTH, "%s.embedding_length" },

0 commit comments

Comments
 (0)