Struct HfHub 

pub struct HfHub { /* private fields */ }

Hugging Face Hub interface

Implementations

impl HfHub

pub fn new() -> Self

Create new HF Hub interface

pub fn with_token(self, token: String) -> Self

Set authentication token

pub fn with_cache_dir<P: AsRef<Path>>(self, cache_dir: P) -> Self

Set cache directory

pub fn list_models(&self, filter: Option<&str>) -> Result<Vec<String>>

List available models

pub fn model_info(&mut self, model_id: &str) -> Result<HfModelInfo>

Get model information

pub fn download_model<P: AsRef<Path>>( &self, model_id: &str, cache_dir: Option<P>, ) -> Result<PathBuf>

Download model files

pub fn upload_model<P: AsRef<Path>>( &self, model_path: P, repo_id: &str, commit_message: Option<&str>, ) -> Result<()>

Upload model to hub

pub fn create_repo(&self, repo_id: &str, private: bool) -> Result<()>

Create model repository

pub fn get_cached_model_path(&self, model_id: &str) -> PathBuf

Get cached model path

Trait Implementations

impl Debug for HfHub

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

Formats the value using the given formatter.

impl Default for HfHub

fn default() -> Self

Returns the “default value” for a type.