Struct MaskPredictor 

pub struct MaskPredictor { /* private fields */ }

Predicts attention masks from prompt information

Implementations

impl MaskPredictor

pub fn new(config: PredictorConfig) -> Self

Create a new predictor with default settings

pub fn predict( &self, prompt: &str, step: Option<u32>, total_steps: Option<u32>, ) -> Prediction

Predict mask from prompt text

pub fn predict_from_embedding( &self, embedding: &[f32], step: Option<u32>, total_steps: Option<u32>, ) -> Prediction

Predict mask from embedding vector (for faster prediction)

pub fn set_total_steps(&mut self, total_steps: usize)

Update step multipliers for different total step counts

pub fn config(&self) -> &PredictorConfig

Get current configuration

Trait Implementations

impl Clone for MaskPredictor

fn clone(&self) -> MaskPredictor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MaskPredictor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.