Struct Config

pub struct Config {
    pub input_names: Vec<String>,
    pub output_name: String,
    pub max_sequence_length: usize,
    pub vocab_size: usize,
    pub model_type: String,
}

Configuration for CoreML models

Fields

input_names: Vec<String>

Input tensor names in order (e.g., [“input_ids”, “token_type_ids”, “attention_mask”])

output_name: String

Output tensor name (e.g., “logits”)

max_sequence_length: usize

Maximum sequence length

vocab_size: usize

Vocabulary size

model_type: String

Model architecture name

Implementations

impl Config

pub fn bert_config( output_name: &str, max_seq_len: usize, vocab_size: usize, ) -> Self

Create BERT-style config with input_ids, token_type_ids, and attention_mask

Trait Implementations

impl Clone for Config

fn clone(&self) -> Config

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Config

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

Formats the value using the given formatter.

impl Default for Config

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for Config

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for Config

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

where __S: Serializer,

Serialize this value into the given Serde serializer.