pub fn predict_native(
encoder_tokens: &[f32],
weights: &Vjepa2PredictorWeights,
cfg: &Vjepa2Config,
batch: usize,
seq: usize,
masks: &Vjepa2Masks,
) -> Result<Vjepa2PredictorOutput, Error>Expand description
Run the predictor on encoder outputs [batch, seq, enc_dim] flat.