llama : adapt plamo to new ffn

ggml-ci
This commit is contained in:
Georgi Gerganov 2023-12-27 17:24:34 +02:00
parent 278f3e99c2
commit 9174699f84
No known key found for this signature in database
GPG key ID: 449E073F9DC10735

View file

@ -5724,6 +5724,7 @@ struct llm_build_context {
model.layers[il].ffn_up, NULL, model.layers[il].ffn_up, NULL,
model.layers[il].ffn_gate, NULL, model.layers[il].ffn_gate, NULL,
model.layers[il].ffn_down, NULL, model.layers[il].ffn_down, NULL,
NULL,
LLM_FFN_SILU, LLM_FFN_PAR, cb, il); LLM_FFN_SILU, LLM_FFN_PAR, cb, il);
cb(cur, "ffn_out", il); cb(cur, "ffn_out", il);
} }