mirror of
https://github.com/jafioti/luminal.git
synced 2026-06-01 21:49:47 +09:00
allow lots of args
This commit is contained in:
@@ -165,6 +165,7 @@ fn swiglu_mlp_ref(
|
|||||||
(gate * up).unwrap().matmul(&w_down.t().unwrap()).unwrap()
|
(gate * up).unwrap().matmul(&w_down.t().unwrap()).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[allow(clippy::too_many_arguments)]
|
||||||
fn transformer_layer_ref(
|
fn transformer_layer_ref(
|
||||||
x: &CandleTensor,
|
x: &CandleTensor,
|
||||||
attn_norm_w: &CandleTensor,
|
attn_norm_w: &CandleTensor,
|
||||||
|
|||||||
Reference in New Issue
Block a user