llama : deci : support ffn-free with attention (#13296)
This commit is contained in:
parent
39e73ae0d6
commit
bc4e1128f7
1 changed files with 1 additions and 1 deletions
|
@ -4792,7 +4792,7 @@ struct llm_build_deci : public llm_graph_context {
|
||||||
}
|
}
|
||||||
|
|
||||||
// FFN-free layer of Llama-3_1-Nemotron-Ultra-253B
|
// FFN-free layer of Llama-3_1-Nemotron-Ultra-253B
|
||||||
if (n_head == 0 && n_ff == 0) {
|
if (n_ff == 0) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue