Struct FamilyRegistry 

pub struct FamilyRegistry { /* private fields */ }

Registry of known model families for detection.

Implementations

impl FamilyRegistry

pub fn new() -> FamilyRegistry

Create an empty registry

pub fn register(&mut self, family: Box<dyn ModelFamily>)

Register a model family

pub fn register_alias( &mut self, hf_pattern: &str, parent_family: &str, ) -> Result<(), String>

Register an HF repo pattern as an alias for an existing family.

Used to unblock derived models that share the parent family’s architecture but have different vocab/repo patterns. e.g., register_alias("codellama/*", "llama") lets codellama checkpoints dispatch through the llama loader.

The pattern is a simple glob with * as suffix wildcard. Returns Err if the parent family is not registered.

pub fn resolve_alias(&self, hf_repo: &str) -> Option<&str>

Resolve an HF repo identifier through registered aliases. Returns the parent family for the first matching alias, or None.

pub fn alias_count(&self) -> usize

Number of registered aliases.

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

Get all registered family names

pub fn get(&self, family_name: &str) -> Option<&dyn ModelFamily>

Look up a family by name

pub fn detect_family(&self, tensor_names: &[&str]) -> Option<&dyn ModelFamily>

Detect model family from tensor names using best-match scoring.

Scores each family by counting how many of its expected tensor patterns (embedding + per-layer for layer 0) match the given tensor names. Returns the family with the highest score, which disambiguates families with overlapping naming conventions (e.g., Qwen2’s bias tensors distinguish it from LLaMA/DeepSeek/Mistral which share the same base naming but lack bias patterns).

pub fn detect_from_model_type( &self, model_type: &str, ) -> Option<&dyn ModelFamily>

Detect model family from HuggingFace model_type string.

pub fn len(&self) -> usize

Number of registered families

pub fn is_empty(&self) -> bool

Check if registry is empty

pub fn detect_from_config_str(body: &str) -> Option<&'static str>

Detect model family from a raw HF config.json body string.

Uses discriminator-field-based dispatch in priority order, mirroring the apr-cookbook architecture-demos detector recipe. Falls through to LlamaForCausalLM as the catch-all for transformer configs that don’t match anything more specific.

Returns the family name (e.g. “llama”, “qwen2”, “phi”) if a discriminator matches, or None if the config is unrecognized. Use get(name) to obtain the registered family object after detection.

This is the high-level entry point most callers want — detect_family requires already-parsed tensor names; detect_from_config_str works directly off the JSON body and surfaces all 18 supported families.

pub fn supported_families() -> Vec<&'static str>

All 18 supported family names tracked by the detector dispatch.

Trait Implementations

impl Debug for FamilyRegistry

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

Formats the value using the given formatter.

impl Default for FamilyRegistry

fn default() -> FamilyRegistry

Returns the “default value” for a type.