scxpand.pretrained.model_registry

Registry for pre-trained models and their metadata.

This module defines the available pre-trained models, their Google Drive links, and associated metadata for automatic download and inference.

Functions

get_pretrained_model_info(model_name)

Get information about a pre-trained model.

Classes

PretrainedModelInfo(name, url, version)

Information about a pre-trained model.

class scxpand.pretrained.model_registry.PretrainedModelInfo(name, url, version)

Information about a pre-trained model.

__init__(name, url, version)

name: str

url: str

version: str

scxpand.pretrained.model_registry.get_pretrained_model_info(model_name)

Get information about a pre-trained model.

Parameters:

model_name (str) – Name of the pre-trained model

Return type:

PretrainedModelInfo

Returns:

PretrainedModelInfo object containing model metadata

Raises:

ValueError – If model_name is not found in registry