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

Fix incorrect check for K norm

Browse source
This commit is contained in:
Ashish 2024-04-14 16:25:55 -07:00
parent bf1a9a5514
commit d2ab693066
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama.cpp
View file

@ -8124,7 +8124,7 @@ struct llm_build_context {
LLM_NORM, cb, il); LLM_NORM, cb, il);
cb(Qcur, "Qcur", il); cb(Qcur, "Qcur", il);
} }
if (model.layers[il].attn_q_norm) { if (model.layers[il].attn_k_norm) {
Kcur = llm_build_norm(ctx0, Kcur, hparams, Kcur = llm_build_norm(ctx0, Kcur, hparams,
model.layers[il].attn_k_norm, model.layers[il].attn_k_norm,
NULL, NULL,
@ -8175,7 +8175,7 @@ struct llm_build_context {
} else { } else {
// parallel residual // parallel residual
cur = inpSA; cur = inpSA;
} }
cur = llm_build_ffn(ctx0, cur, cur = llm_build_ffn(ctx0, cur,
model.layers[il].ffn_up, NULL, model.layers[il].ffn_up, NULL,
model.layers[il].ffn_gate, NULL, model.layers[il].ffn_gate, NULL,