Struct ModelRegistry 

pub struct ModelRegistry<B: Backend> { /* private fields */ }

Maps architecture identifiers (config.json::model_type, plus known aliases) to loaders. Mirrors MLC’s model.py::MODELS table.

Implementations

impl<B: Backend> ModelRegistry<B>

pub fn new() -> Self

Creates a registry with the built-in architectures registered.

pub fn register( &mut self, architecture: &str, loader: fn(&dyn ModelSource, &Device<B>) -> Result<Box<dyn GenerativeModel<B>>>, )

Registers (or replaces) the loader for an architecture id.

pub fn supports(&self, architecture: &str) -> bool

Whether an architecture id has a loader.

pub fn architectures(&self) -> Vec<&str>

Registered architecture ids.

pub fn load( &self, source: &dyn ModelSource, device: &Device<B>, ) -> Result<Box<dyn GenerativeModel<B>>>

Loads the model described by source’s metadata.

Trait Implementations

impl<B: Backend> Default for ModelRegistry<B>

fn default() -> Self

Returns the “default value” for a type.