Oops

EricLBuehler · Aug 21, 2024 · c0b93f9 · c0b93f9
1 parent f12c9a9
commit c0b93f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mistralrs-core/src/models/mamba2.rs b/mistralrs-core/src/models/mamba2.rs
@@ -145,7 +145,7 @@ impl Mixer {
             cfg.hidden_size,
             cfg.use_bias,
             &cfg.quantization_config,
-            vb.pp("in_proj"),
+            vb.pp("out_proj"),
         )?;
 
         // Time step proj, discretization