From 6beebf3fd92aa2ab5c8b060b6d25c5e1d3d12e95 Mon Sep 17 00:00:00 2001 From: klosax <131523366+klosax@users.noreply.github.com> Date: Sun, 13 Aug 2023 14:11:01 +0200 Subject: [PATCH] gptneox-main.cpp : add file_type key --- gptneox-main.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/gptneox-main.cpp b/gptneox-main.cpp index 4773dfd69..63ee5e61c 100644 --- a/gptneox-main.cpp +++ b/gptneox-main.cpp @@ -379,6 +379,8 @@ bool gpt_neox_model_load(const std::string & fname, gpt_neox_model & model, gpt2 if (keyidx != -1) { fprintf(stdout, "%s: model license = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } keyidx = gguf_find_key(ggufctx, "general.architecture"); if (keyidx != -1) { fprintf(stdout, "%s: model architecture = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } + keyidx = gguf_find_key(ggufctx, "general.file_type"); + if (keyidx != -1) { fprintf(stdout, "%s: model file type = %s\n", __func__, gguf_get_val_str(ggufctx, keyidx)); } } // check required metadata