Struct RlxVisionModel 

pub struct RlxVisionModel { /* private fields */ }

RLX-compiled NomicVision encoder (patch preprocess host-side, trunk on RLX).

Implementations

impl RlxVisionModel

pub fn load_sized( config_path: &Path, weights_path: &str, batch: usize, ) -> Result<Self>

pub fn load_sized_on( config_path: &Path, weights_path: &str, batch: usize, device: Device, ) -> Result<Self>

pub fn forward(&mut self, pixel_values: &[f32], batch: usize) -> Vec<f32>

Forward: pixel_values [batch, 3, img, img] row-major NCHW → CLS [batch, hidden].

pub fn forward_all( &mut self, pixel_values: &[f32], batch: usize, ) -> Vec<Vec<f32>>

pub fn forward_slots(&mut self, hidden: &[f32]) -> (*const f32, usize)

pub fn hidden_size(&self) -> usize

pub fn img_size(&self) -> usize

pub fn patch_size(&self) -> usize

pub fn num_patches(&self) -> usize

pub fn preprocess_weights(&self) -> &VisionPreprocessWeights