Skip to content

Commit bff1b65

Browse files
committed
add nightly requirement for mamba-minimal
1 parent cac2f33 commit bff1b65

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

dfdx/src/nn/layers/mod.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ mod leaky_relu;
2525
mod linear;
2626
mod ln;
2727
mod log_softmax;
28+
#[cfg(feature = "nightly")]
2829
mod mamba_minimal;
2930
mod matmul;
3031
mod multi_head_attention;
@@ -80,6 +81,7 @@ pub use leaky_relu::LeakyReLU;
8081
pub use linear::{Linear, LinearConfig, LinearConstConfig};
8182
pub use ln::Ln;
8283
pub use log_softmax::LogSoftmax;
84+
#[cfg(feature = "nightly")]
8385
pub use mamba_minimal::{
8486
stateful::{MambaStateCache, MambaStateCacheConfig},
8587
MambaBlock, MambaBlockConfig, MambaBlockConstConfig,

0 commit comments

Comments
 (0)