summaryrefslogtreecommitdiff
path: root/examples/gptneox-wip/falcon-main.cpp
diff options
context:
space:
mode:
authorCebtenzzre <cebtenzzre@gmail.com>2023-09-27 12:18:07 -0400
committerGitHub <noreply@github.com>2023-09-27 12:18:07 -0400
commit20c7e1e804690f3db58bd33eb56f8c6aa4735c63 (patch)
tree5c768feefd0e547085c620811a2fb56ad642933c /examples/gptneox-wip/falcon-main.cpp
parentdc6897404e141c74cbbf8030ecfebd74e1815411 (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.cpp4
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)); }
}