vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gptneox-main.cpp : n_layer --> n_block

Browse source
This commit is contained in:
klosax 2023-08-13 00:03:32 +02:00 • committed by GitHub
parent e91a2224e4
commit c7bd8c147c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
gptneox-main.cpp
View file

@ -414,7 +414,7 @@ bool gpt_neox_model_load(const std::string & fname, gpt_neox_model & model, gpt2
if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.attention.head_count"); if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.attention.head_count");
if (keyidx != -1) { hparams.n_head = gguf_get_val_u32(ggufctx, keyidx); } else { ok = false; } } if (keyidx != -1) { hparams.n_head = gguf_get_val_u32(ggufctx, keyidx); } else { ok = false; } }
if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.layer_count"); if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.block_count");
if (keyidx != -1) { hparams.n_block = gguf_get_val_u32(ggufctx, keyidx); } else { ok = false; } } if (keyidx != -1) { hparams.n_block = gguf_get_val_u32(ggufctx, keyidx); } else { ok = false; } }
if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.rope.dimension_count"); if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.rope.dimension_count");