diff options
author | Cebtenzzre <cebtenzzre@gmail.com> | 2023-09-27 12:18:07 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-27 12:18:07 -0400 |
commit | 20c7e1e804690f3db58bd33eb56f8c6aa4735c63 (patch) | |
tree | 5c768feefd0e547085c620811a2fb56ad642933c /examples/gptneox-wip/falcon-main.cpp | |
parent | dc6897404e141c74cbbf8030ecfebd74e1815411 (diff) |
gguf : fix a few general keys (#3341)
Diffstat (limited to 'examples/gptneox-wip/falcon-main.cpp')
-rw-r--r-- | examples/gptneox-wip/falcon-main.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/gptneox-wip/falcon-main.cpp b/examples/gptneox-wip/falcon-main.cpp index 7f9a1620..e9197f6b 100644 --- a/examples/gptneox-wip/falcon-main.cpp +++ b/examples/gptneox-wip/falcon-main.cpp @@ -367,10 +367,10 @@ bool falcon_model_load(const std::string & fname, falcon_model & model, gpt2bpe_ keyidx = gguf_find_key(ggufctx, "general.architecture"); if (keyidx != -1) { printf("%s: model architecture = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } keyidx = gguf_find_key(ggufctx, "general.file_type"); - if (keyidx != -1) { printf("%s: model file type = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } + if (keyidx != -1) { printf("%s: model file type = %" PRIu32 "\n", __func__, gguf_get_val_u32(ggufctx, keyidx)); } keyidx = gguf_find_key(ggufctx, "gptneox.tensor_data_layout"); if (keyidx != -1) { printf("%s: model data layout = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } - keyidx = gguf_find_key(ggufctx, "general.source.hugginface.repository"); + keyidx = gguf_find_key(ggufctx, "general.source.huggingface.repository"); if (keyidx != -1) { printf("%s: model source HF repo = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } } |