Struct ModelHub 

pub struct ModelHub { /* private fields */ }

Model hub for downloading and caching pretrained weights.

Implementations

impl ModelHub

pub fn new() -> Self

Creates a new hub with the default cache directory and built-in registry of known pretrained models.

pub fn cache_dir(&self) -> &Path

Returns the cache directory path.

pub fn registry(&self) -> &HashMap<String, HubEntry>

Returns a reference to the internal registry.

pub fn download_if_missing(&self, name: &str) -> Result<PathBuf, ModelError>

Ensures the weight file for name is present in the local cache, downloading it via curl if necessary.

Returns the path to the cached file on success.

pub fn load_weights( &self, name: &str, ) -> Result<HashMap<String, Tensor>, ModelError>

Downloads (if needed) and loads all tensors from the safetensors weight file for the given model name.

Trait Implementations

impl Default for ModelHub

fn default() -> Self

Returns the “default value” for a type.