File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
candle-transformers/src/models/smol Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -11,7 +11,7 @@ const MAX_SEQ_LEN: usize = 4096;
1111use candle:: IndexOp ;
1212
1313// ===== RECONSTRUCTION FUNCTION =====
14- fn reconstruct_qk_weights ( gguf_weight : & Tensor , num_heads : usize ) -> Result < Tensor > {
14+ fn reconstruct_qk_weights ( gguf_weight : & Tensor , _num_heads : usize ) -> Result < Tensor > {
1515 let total_rows = gguf_weight. dim ( 0 ) ?;
1616 let half_rows = total_rows / 2 ;
1717 let chunk_size = 128 ;
Original file line number Diff line number Diff line change @@ -289,7 +289,7 @@ impl SmolLM3Attention {
289289}
290290
291291#[ derive( Debug , Clone ) ]
292- struct DecoderLayer {
292+ pub ( crate ) struct DecoderLayer {
293293 self_attn : SmolLM3Attention ,
294294 mlp : SmolLM3MLP ,
295295 ln1 : RmsNorm ,
You can’t perform that action at this time.
0 commit comments