Struct Config 

pub struct Config { /* private fields */ }

Configuration for a Model. See Model::from_pretrained for a simple way to get started.

Implementations

impl Config

pub fn transducer( encoder: &str, decoder: &str, joiner: &str, tokens: &str, ) -> Self

Make a new Config for a transducer model with reasonable defaults.

pub fn paraformer(encoder: &str, decoder: &str, tokens: &str) -> Self

Make a new Config for a paraformer model with reasonable defaults.

pub fn zipformer2_ctc(model: &str, tokens: &str) -> Self

Make a new Config for a zipformer2 ctc model with reasonable defaults.

pub fn sample_rate(self, rate: usize) -> Self

Set the model’s sample rate - usually 16000 for most transducers.

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

Set the model’s feature dimension - usually 80 for most transducers.

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

Set the number of threads to use. Defaults to physical core count or 8, whichever is smaller.

pub fn cpu(self) -> Self

Use CPU as the compute provider.

pub fn cuda(self) -> Self

Available on crate feature cuda only.

Use CUDA as the compute provider. This requires CUDA 11.8.

pub fn directml(self) -> Self

Available on crate feature directml only.

Use DirectML as the compute provider.

pub fn debug(self, enable: bool) -> Self

Print debug messages at model load time.

pub fn greedy_search(self) -> Self

Take the symbol with largest posterior probability of each frame as the decoding result.

pub fn modified_beam_search(self) -> Self

Keep topk states for each frame, then expand kept states with their own contexts to next frame.

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

Maximum number of active paths to keep when Config::modified_beam_search is used.

Defaults to 16.

pub fn detect_endpoints(self, enable: bool) -> Self

Enable endpoint detection. Defaults to disabled.

pub fn rule1_min_trailing_silence(self, seconds: f32) -> Self

Detect endpoint if trailing silence is larger than this value even if nothing has been decoded.

pub fn rule2_min_trailing_silence(self, seconds: f32) -> Self

Detect endpoint if trailing silence is larger than this value and a non-blank has been decoded.

pub fn rule3_min_utterance_length(self, seconds: f32) -> Self

Detect an endpoint if an utterance is larger than this value.

pub fn build(self) -> Result<Model>

Build your very own Model.

Trait Implementations

impl Clone for Config

fn clone(&self) -> Config

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.