Fix incorrect check for K norm

This commit is contained in:
Ashish 2024-04-14 16:25:55 -07:00
parent bf1a9a5514
commit d2ab693066

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,