feat: support HF_ENDPOINT environment when downloading model (#505)

Author: StrayDragon

Cargo.lock

@@ -29,7 +29,7 @@ memmap2 = "^0.9"
 [dev-dependencies]
 insta = { git = "https://github.com/OlivierDehaene/insta", rev = "f4f98c0410b91fb5a28b10df98e4422955be9c2c", features = ["yaml"] }
 is_close = "0.1.3"
-hf-hub = "0.4.2"
+hf-hub = { workspace = true }
 anyhow = { workspace = true }
 tokenizers = { workspace = true }
 serial_test = { workspace = true }

backends/candle/Cargo.toml

@@ -107,7 +107,7 @@ pub fn download_artifacts(
     model_id: &'static str,
     revision: Option<&'static str>,
 ) -> Result<PathBuf> {
-    let mut builder = ApiBuilder::new().with_progress(false);
+    let mut builder = ApiBuilder::from_env().with_progress(false);
 
     if let Some(cache_dir) = std::env::var_os("HUGGINGFACE_HUB_CACHE") {
         builder = builder.with_cache_dir(cache_dir.into());

backends/candle/tests/common.rs

@@ -70,7 +70,9 @@ pub async fn run(
         // Using a local model
         (model_id_path.to_path_buf(), None)
     } else {
-        let mut builder = ApiBuilder::new().with_progress(false).with_token(hf_token);
+        let mut builder = ApiBuilder::from_env()
+            .with_progress(false)
+            .with_token(hf_api_token);
 
         if let Some(cache_dir) = huggingface_hub_cache {
             builder = builder.with_cache_dir(cache_dir.into());