Struct HuggingFaceLoader 

pub struct HuggingFaceLoader {
    pub hf_token: Option<String>,
    pub hf_token_env_var: String,
    pub hf_api: OnceCell<Api>,
}

Fields

hf_token: Option<String>

hf_token_env_var: String

hf_api: OnceCell<Api>

Implementations

impl HuggingFaceLoader

pub fn new() -> Self

pub fn hf_api(&self) -> &Api

pub fn load_file<T: AsRef<str>, S: Into<String>>( &self, file_name: T, repo_id: S, ) -> Result<PathBuf>

pub fn load_model_safe_tensors<S: Into<String>>( &self, repo_id: S, ) -> Result<Vec<PathBuf>>

pub fn canonicalize_local_path(local_path: PathBuf) -> Result<PathBuf>

pub fn parse_full_model_url(model_url: &str) -> (String, String, String)

pub fn model_url_from_repo_and_local_filename( repo_id: &str, local_model_filename: &str, ) -> String

pub fn model_url_from_repo(repo_id: &str) -> String

pub fn model_id_from_url(model_url: &str) -> String

Trait Implementations

impl Clone for HuggingFaceLoader

fn clone(&self) -> HuggingFaceLoader

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for HuggingFaceLoader

fn default() -> Self

Returns the “default value” for a type.

impl HfTokenTrait for HuggingFaceLoader

fn hf_token_mut(&mut self) -> &mut Option<String>

fn hf_token_env_var_mut(&mut self) -> &mut String

fn hf_token<S: Into<String>>(self, hf_token: S) -> Self

where Self: Sized,

Set the API key for the client. Otherwise it will attempt to load it from the .env file.

fn hf_token_env_var<S: Into<String>>(self, hf_token_env_var: S) -> Self

where Self: Sized,

Set the environment variable name for the API key. Default is “HUGGING_FACE_TOKEN”.