Struct SparseMaskBuilder 

pub struct SparseMaskBuilder { /* private fields */ }

Builder for creating sparse masks

Implementations

impl SparseMaskBuilder

pub fn new(n: usize) -> Self

pub fn with_local_window(self, window_size: usize) -> Self

Add local window pattern

pub fn with_global_tokens(self, global_indices: &[usize]) -> Self

Add global tokens (all positions attend to these)

pub fn with_causal(self) -> Self

Add causal masking

pub fn build(self) -> AttentionMask

Build the mask