Skip to main content

BlockDenoiseSampler

rlx_models::tide

Trait BlockDenoiseSampler 

Source 
pub trait BlockDenoiseSampler {
    // Required method
    fn sample_block(
        &mut self,
        x: &[u32],
        window_end: usize,
        block_length: usize,
        refresh_experts: bool,
        gen_cfg: &GenerateConfig,
        model_cfg: &LLaDA2MoeConfig,
        step_ctx: DenoiseStepCtx,
    ) -> Result<(Vec<u32>, Vec<f32>), Error>;
}
Expand description

Sample (x0, x0_p) for the trailing block of the active window.

Required Methods§

Source

fn sample_block( &mut self, x: &[u32], window_end: usize, block_length: usize, refresh_experts: bool, gen_cfg: &GenerateConfig, model_cfg: &LLaDA2MoeConfig, step_ctx: DenoiseStepCtx, ) -> Result<(Vec<u32>, Vec<f32>), Error>

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§