Struct ModelBuilder 

pub struct ModelBuilder { /* private fields */ }

Builder for common model configurations

Implementations

impl ModelBuilder

pub fn mamba2() -> Self

Start building a Mamba2 model

pub fn rwkv() -> Self

Start building an RWKV model

pub fn s4() -> Self

Start building an S4 model

pub fn s4d() -> Self

Start building an S4D model

pub fn transformer() -> Self

Start building a Transformer model

pub fn dims(self, input: usize, hidden: usize, output: usize) -> Self

Set dimensions

pub fn layers(self, n: usize) -> Self

Set number of layers

pub fn state_dim(self, dim: usize) -> Self

Set state dimension

pub fn build(self) -> ModelConfig

Build the configuration