Struct RlxEmbed 

pub struct RlxEmbed { /* private fields */ }

High-level embedding model — auto-detects BERT / NomicBERT / NomicVision.

Implementations

impl RlxEmbed

pub fn from_dir(dir: &Path, pooling: Pooling) -> Result<Self>

Load from a local directory (config.json + model.safetensors) on CPU.

pub fn from_dir_on(dir: &Path, pooling: Pooling, device: Device) -> Result<Self>

Load from a local directory on the given device.

pub fn from_weights(path: &Path, pooling: Pooling) -> Result<Self>

Load from a .gguf file or a directory containing one (optional sidecar config.json).

pub fn from_weights_on( path: &Path, pooling: Pooling, device: Device, ) -> Result<Self>

Load weights path on the given device.

pub fn device(&self) -> Device

Execution device for this instance.

pub fn dim(&self) -> usize

pub fn arch(&self) -> Arch

pub fn forward( &mut self, inputs: &[(&str, &[f32])], batch: usize, seq: usize, ) -> Result<Vec<f32>>

Forward on pre-tokenized inputs; returns flattened hidden states.