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);
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,
model.layers[il].attn_k_norm,
NULL,
@ -8175,7 +8175,7 @@ struct llm_build_context {
} else {
// parallel residual
cur = inpSA;
}
}
cur = llm_build_ffn(ctx0, cur,
model.layers[il].ffn_up, NULL,
model.layers[il].ffn_gate, NULL,