added builder with_endpoint to tokio.rs

huggingface · Dec 13, 2024 · 9cf016a · 9cf016a
1 parent e7d5f74
commit 9cf016a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/api/tokio.rs b/src/api/tokio.rs
@@ -130,6 +130,12 @@ impl ApiBuilder {
         self.progress = progress;
         self
     }
+
+    /// Changes the endpoint of the API. Default is `https://huggingface.co`.
+    pub fn with_endpoint(mut self, endpoint: String) -> Self {
+        self.endpoint = endpoint;
+        self
+    }
 
     /// Changes the location of the cache directory. Defaults is `~/.cache/huggingface/`.
     pub fn with_cache_dir(mut self, cache_dir: PathBuf) -> Self {