Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Ashish 2024-04-14 16:24:00 -07:00 committed by GitHub
parent 91728faac6
commit bf1a9a5514
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -8177,11 +8177,11 @@ struct llm_build_context {
cur = inpSA;
}
cur = llm_build_ffn(ctx0, cur,
model.layers[il].ffn_up, NULL,
model.layers[il].ffn_gate, NULL,
model.layers[il].ffn_down, NULL,
NULL,
LLM_FFN_SILU, LLM_FFN_PAR, cb, il);
model.layers[il].ffn_up, NULL,
model.layers[il].ffn_gate, NULL,
model.layers[il].ffn_down, NULL,
NULL,
LLM_FFN_SILU, LLM_FFN_PAR, cb, il);
cb(cur, "ffn_out", il);
}