Skip to main content

ModelFamilyImpl

trustformers_models::common_patterns

Trait ModelFamilyImpl 

Source 
pub trait ModelFamilyImpl: Send + Sync {
    // Required methods
    fn create_config(
        &self,
        size: &str,
        variant: Option<&str>,
    ) -> Result<Box<dyn DynConfig>>;
    fn family_name(&self) -> &'static str;
    fn available_sizes(&self) -> Vec<&'static str>;
    fn available_variants(&self) -> Vec<&'static str>;
    fn use_cases(&self) -> Vec<&'static str>;
    fn metadata(&self) -> ModelFamilyMetadata;
}
Expand description

Model family implementation for registry

Required Methods§

Source

fn create_config( &self, size: &str, variant: Option<&str>, ) -> Result<Box<dyn DynConfig>>

Create configuration for a specific size and variant

Source

fn family_name(&self) -> &'static str

Get family name

Source

fn available_sizes(&self) -> Vec<&'static str>

Get available model sizes

Source

fn available_variants(&self) -> Vec<&'static str>

Get available variants (base, instruct, chat, etc.)

Source

fn use_cases(&self) -> Vec<&'static str>

Get recommended use cases for this family

Source

fn metadata(&self) -> ModelFamilyMetadata

Get model family metadata

Implementors§