Struct ModelConfig 

pub struct ModelConfig {
    pub model_path: PathBuf,
    pub tokenizer_path: Option<PathBuf>,
    pub dimension: Option<usize>,
    pub max_length: usize,
}

Configuration for loading an embedding model.

Fields

model_path: PathBuf

Path to model directory or ONNX file.

tokenizer_path: Option<PathBuf>

Path to tokenizer.json (optional if in model_path).

dimension: Option<usize>

Embedding dimension (auto-detect if None).

max_length: usize

Maximum sequence length.

Implementations

impl ModelConfig

pub fn from_directory<P: AsRef<Path>>(path: P) -> Self

Create config for a model directory.

pub fn with_max_length(self, max_length: usize) -> Self

Set maximum sequence length.

pub fn with_dimension(self, dimension: usize) -> Self

Set embedding dimension.

pub fn load(&self) -> Result<OnnxEmbedder>

Load the embedder from this config.

Trait Implementations

impl Clone for ModelConfig

fn clone(&self) -> ModelConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelConfig

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

Formats the value using the given formatter.

impl Default for ModelConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ModelConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ModelConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.