Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Model registry for creating and managing model instances

Implementations

impl ModelRegistry

pub fn new() -> Self

Create a new model registry

pub fn register(&mut self, name: impl Into<String>, config: ModelConfig)

Register a model configuration with a name

pub fn get_config(&self, name: &str) -> Option<&ModelConfig>

Get a registered configuration

pub fn list_models(&self) -> Vec<String>

List all registered model names

pub fn create_model( &self, name: &str, ) -> InferenceResult<Box<dyn AutoregressiveModel>>

Create a model wrapper from a configuration

pub fn load_weights( &self, _model: &mut dyn AutoregressiveModel, path: impl AsRef<Path>, ) -> InferenceResult<()>

Load weights from a file

Trait Implementations

impl Default for ModelRegistry

fn default() -> Self

Returns the “default value” for a type.