huggingface-hub 0.30.0rc2__py3-none-any.whl → 0.30.1__py3-none-any.whl

huggingface_hub/__init__.py

@@ -46,7 +46,7 @@ import sys
 from typing import TYPE_CHECKING
 
 
-__version__ = "0.30.0rc2"
+__version__ = "0.30.1"
 
 # Alphabetical order of definitions is ensured in tests
 # WARNING: any comment added in this dictionary definition will be lost when

huggingface_hub/constants.py

@@ -73,6 +73,7 @@ if _staging_mode:
 HUGGINGFACE_HEADER_X_REPO_COMMIT = "X-Repo-Commit"
 HUGGINGFACE_HEADER_X_LINKED_ETAG = "X-Linked-Etag"
 HUGGINGFACE_HEADER_X_LINKED_SIZE = "X-Linked-Size"
+HUGGINGFACE_HEADER_X_BILL_TO = "X-HF-Bill-To"
 
 INFERENCE_ENDPOINT = os.environ.get("HF_INFERENCE_ENDPOINT", "https://api-inference.huggingface.co")
 

huggingface_hub/inference/_client.py

@@ -146,6 +146,9 @@ class InferenceClient:
         headers (`Dict[str, str]`, `optional`):
             Additional headers to send to the server. By default only the authorization and user-agent headers are sent.
             Values in this dictionary will override the default values.
+        bill_to (`str`, `optional`):
+            The billing account to use for the requests. By default the requests are billed on the user's account.
+            Requests can only be billed to an organization the user is a member of, and which has subscribed to Enterprise Hub.
         cookies (`Dict[str, str]`, `optional`):
             Additional cookies to send to the server.
         proxies (`Any`, `optional`):
@@ -168,6 +171,7 @@ class InferenceClient:
         headers: Optional[Dict[str, str]] = None,
         cookies: Optional[Dict[str, str]] = None,
         proxies: Optional[Any] = None,
+        bill_to: Optional[str] = None,
         # OpenAI compatibility
         base_url: Optional[str] = None,
         api_key: Optional[str] = None,
@@ -203,7 +207,25 @@ class InferenceClient:
 
         self.model: Optional[str] = base_url or model
         self.token: Optional[str] = token
-        self.headers = headers if headers is not None else {}
+
+        self.headers = {**headers} if headers is not None else {}
+        if bill_to is not None:
+            if (
+                constants.HUGGINGFACE_HEADER_X_BILL_TO in self.headers
+                and self.headers[constants.HUGGINGFACE_HEADER_X_BILL_TO] != bill_to
+            ):
+                warnings.warn(
+                    f"Overriding existing '{self.headers[constants.HUGGINGFACE_HEADER_X_BILL_TO]}' value in headers with '{bill_to}'.",
+                    UserWarning,
+                )
+            self.headers[constants.HUGGINGFACE_HEADER_X_BILL_TO] = bill_to
+
+            if token is not None and not token.startswith("hf_"):
+                warnings.warn(
+                    "You've provided an external provider's API key, so requests will be billed directly by the provider. "
+                    "The `bill_to` parameter is only applicable for Hugging Face billing and will be ignored.",
+                    UserWarning,
+                )
 
         # Configure provider
         self.provider = provider if provider is not None else "hf-inference"

huggingface_hub/inference/_generated/_async_client.py

@@ -134,6 +134,9 @@ class AsyncInferenceClient:
         headers (`Dict[str, str]`, `optional`):
             Additional headers to send to the server. By default only the authorization and user-agent headers are sent.
             Values in this dictionary will override the default values.
+        bill_to (`str`, `optional`):
+            The billing account to use for the requests. By default the requests are billed on the user's account.
+            Requests can only be billed to an organization the user is a member of, and which has subscribed to Enterprise Hub.
         cookies (`Dict[str, str]`, `optional`):
             Additional cookies to send to the server.
         trust_env ('bool', 'optional'):
@@ -159,6 +162,7 @@ class AsyncInferenceClient:
         cookies: Optional[Dict[str, str]] = None,
         trust_env: bool = False,
         proxies: Optional[Any] = None,
+        bill_to: Optional[str] = None,
         # OpenAI compatibility
         base_url: Optional[str] = None,
         api_key: Optional[str] = None,
@@ -194,7 +198,25 @@ class AsyncInferenceClient:
 
         self.model: Optional[str] = base_url or model
         self.token: Optional[str] = token
-        self.headers = headers if headers is not None else {}
+
+        self.headers = {**headers} if headers is not None else {}
+        if bill_to is not None:
+            if (
+                constants.HUGGINGFACE_HEADER_X_BILL_TO in self.headers
+                and self.headers[constants.HUGGINGFACE_HEADER_X_BILL_TO] != bill_to
+            ):
+                warnings.warn(
+                    f"Overriding existing '{self.headers[constants.HUGGINGFACE_HEADER_X_BILL_TO]}' value in headers with '{bill_to}'.",
+                    UserWarning,
+                )
+            self.headers[constants.HUGGINGFACE_HEADER_X_BILL_TO] = bill_to
+
+            if token is not None and not token.startswith("hf_"):
+                warnings.warn(
+                    "You've provided an external provider's API key, so requests will be billed directly by the provider. "
+                    "The `bill_to` parameter is only applicable for Hugging Face billing and will be ignored.",
+                    UserWarning,
+                )
 
         # Configure provider
         self.provider = provider if provider is not None else "hf-inference"

huggingface_hub/inference/_providers/hf_inference.py

@@ -151,6 +151,14 @@ def _check_supported_task(model: str, task: str) -> None:
             return  # Only conversational allowed if tagged as conversational
         raise ValueError("Non-conversational image-text-to-text task is not supported.")
 
+    if (
+        task in ("feature-extraction", "sentence-similarity")
+        and pipeline_tag in ("feature-extraction", "sentence-similarity")
+        and task in tags
+    ):
+        # feature-extraction and sentence-similarity are interchangeable for HF Inference
+        return
+
     # For all other tasks, just check pipeline tag
     if pipeline_tag != task:
         raise ValueError(

{huggingface_hub-0.30.0rc2.dist-info → huggingface_hub-0.30.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface-hub
-Version: 0.30.0rc2
+Version: 0.30.1
 Summary: Client library to download and publish models, datasets and other repos on the huggingface.co hub
 Home-page: https://github.com/huggingface/huggingface_hub
 Author: Hugging Face, Inc.

{huggingface_hub-0.30.0rc2.dist-info → huggingface_hub-0.30.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-huggingface_hub/__init__.py,sha256=I04J88IWTa5PNrze79AZCmXiKl4p6MJZ9J7vmzEvWD4,49371
+huggingface_hub/__init__.py,sha256=hNa8rkJtZJkpw6ZIpTkY6if38nViltBFbMDSy3nhpj4,49368
 huggingface_hub/_commit_api.py,sha256=cihZ2Zn4_AEjSVSuxwN7Sr42aKz2a38OkBFSqD-JB7I,38777
 huggingface_hub/_commit_scheduler.py,sha256=tfIoO1xWHjTJ6qy6VS6HIoymDycFPg0d6pBSZprrU2U,14679
 huggingface_hub/_inference_endpoints.py,sha256=SLoZOQtv_hNl0Xuafo34L--zuCZ3zSJja2tSkYkG5V4,17268
@@ -11,7 +11,7 @@ huggingface_hub/_upload_large_folder.py,sha256=eedUTowflZx1thFVLDv7hLd_LQqixa5NV
 huggingface_hub/_webhooks_payload.py,sha256=Xm3KaK7tCOGBlXkuZvbym6zjHXrT1XCrbUFWuXiBmNY,3617
 huggingface_hub/_webhooks_server.py,sha256=5J63wk9MUGKBNJVsOD9i60mJ-VMp0YYmlf87vQsl-L8,15767
 huggingface_hub/community.py,sha256=4MtcoxEI9_0lmmilBEnvUEi8_O1Ivfa8p6eKxYU5-ts,12198
-huggingface_hub/constants.py,sha256=PgTmJ8f8PegpXBBaXoULUYTDnn7pHW9XlTB9vtZxSi8,9435
+huggingface_hub/constants.py,sha256=4rn5JWp4k5JRNWcnl1XkPPFr-d6GkYtkkg0-qLcxcj4,9481
 huggingface_hub/errors.py,sha256=cE0bwLHbv8e34tbOdlkl-exjDoFxGgCLYmTYlDkpJgI,10155
 huggingface_hub/fastai_utils.py,sha256=DpeH9d-6ut2k_nCAAwglM51XmRmgfbRe2SPifpVL5Yk,16745
 huggingface_hub/file_download.py,sha256=s3kUdf7NhcRjoJpLcYRZOz1deC0Q7k-_gWIBwcd2MG4,76327
@@ -40,10 +40,10 @@ huggingface_hub/commands/upload_large_folder.py,sha256=P-EO44JWVl39Ax4b0E0Z873d0
 huggingface_hub/commands/user.py,sha256=M6Ef045YcyV4mFCbLaTRPciQDC6xtV9MMheeen69D0E,11168
 huggingface_hub/commands/version.py,sha256=vfCJn7GO1m-DtDmbdsty8_RTVtnZ7lX6MJsx0Bf4e-s,1266
 huggingface_hub/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-huggingface_hub/inference/_client.py,sha256=uRNW47cLJwwicE6wxJlJhKvv7uLDOIvqJGFkFlgE5tE,161457
+huggingface_hub/inference/_client.py,sha256=9QydbGgybtFeAsqxFVCG-Usb8GIUcK2lAU0xfQ_Tl_8,162678
 huggingface_hub/inference/_common.py,sha256=iwCkq2fWE1MVoPTeeXN7UN5FZi7g5fZ3K8PHSOCi5dU,14591
 huggingface_hub/inference/_generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-huggingface_hub/inference/_generated/_async_client.py,sha256=5JeW50ZSj4GhZjF3eugZverLCwB81AGxolYWuweLZOY,167662
+huggingface_hub/inference/_generated/_async_client.py,sha256=k20fioyWjC7B8UohguFZSIp6GR-CbJgp3fCvNhgzMGE,168883
 huggingface_hub/inference/_generated/types/__init__.py,sha256=--r3nBmBRtgyQR9TkdQl53_crcKYJhWfTkFK2VE2gUk,6307
 huggingface_hub/inference/_generated/types/audio_classification.py,sha256=Jg3mzfGhCSH6CfvVvgJSiFpkz6v4nNA0G4LJXacEgNc,1573
 huggingface_hub/inference/_generated/types/audio_to_audio.py,sha256=2Ep4WkePL7oJwcp5nRJqApwviumGHbft9HhXE9XLHj4,891
@@ -84,7 +84,7 @@ huggingface_hub/inference/_providers/cerebras.py,sha256=YT1yFhXvDJiKZcqcJcA_7VZJ
 huggingface_hub/inference/_providers/cohere.py,sha256=GkFsuKSaqsyfeerPx0ewv-EX44MtJ8a3XXEfmAiTpb0,419
 huggingface_hub/inference/_providers/fal_ai.py,sha256=Pko6CHSB30jkuuxBNC6rpDWWWN07lvPCNxxIlteBtik,6099
 huggingface_hub/inference/_providers/fireworks_ai.py,sha256=6uDsaxJRaN2xWNQX8u1bvF8zO-8J31TAnHdsrf_TO5g,337
-huggingface_hub/inference/_providers/hf_inference.py,sha256=1vhEpgBbPF1JEcurm_5PqEQaTs2XoHrrnvudT6LJgdU,6865
+huggingface_hub/inference/_providers/hf_inference.py,sha256=s-mHKGN1BQ1gb-l4wYfYZPZW6voJ9XhdNG55ZYYpUoI,7148
 huggingface_hub/inference/_providers/hyperbolic.py,sha256=ZIwD50fUv3QnL091XlKA7TOs_E33Df5stsewpnbDNZ4,1824
 huggingface_hub/inference/_providers/nebius.py,sha256=VcgjvQCNnFifU4GMFXJENgdqgQXSu4uHcu2hUljNDaY,1593
 huggingface_hub/inference/_providers/novita.py,sha256=_ofZzGxtKn9v9anmS1kElnOiJa9Hmoeyi4Exv_jVOVg,2023
@@ -127,9 +127,9 @@ huggingface_hub/utils/insecure_hashlib.py,sha256=OjxlvtSQHpbLp9PWSrXBDJ0wHjxCBU-
 huggingface_hub/utils/logging.py,sha256=0A8fF1yh3L9Ka_bCDX2ml4U5Ht0tY8Dr3JcbRvWFuwo,4909
 huggingface_hub/utils/sha.py,sha256=OFnNGCba0sNcT2gUwaVCJnldxlltrHHe0DS_PCpV3C4,2134
 huggingface_hub/utils/tqdm.py,sha256=xAKcyfnNHsZ7L09WuEM5Ew5-MDhiahLACbbN2zMmcLs,10671
-huggingface_hub-0.30.0rc2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-huggingface_hub-0.30.0rc2.dist-info/METADATA,sha256=FlsJn0Cep50BEx9Kap6HVuRXxw1mXPjuHP9gPpJl6T0,13554
-huggingface_hub-0.30.0rc2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-huggingface_hub-0.30.0rc2.dist-info/entry_points.txt,sha256=Y3Z2L02rBG7va_iE6RPXolIgwOdwUFONyRN3kXMxZ0g,131
-huggingface_hub-0.30.0rc2.dist-info/top_level.txt,sha256=8KzlQJAY4miUvjAssOAJodqKOw3harNzuiwGQ9qLSSk,16
-huggingface_hub-0.30.0rc2.dist-info/RECORD,,
+huggingface_hub-0.30.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+huggingface_hub-0.30.1.dist-info/METADATA,sha256=yb4jdtYon84q0whgsLOLbLf2Z-2HIN6tsp3jckuFBz4,13551
+huggingface_hub-0.30.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+huggingface_hub-0.30.1.dist-info/entry_points.txt,sha256=Y3Z2L02rBG7va_iE6RPXolIgwOdwUFONyRN3kXMxZ0g,131
+huggingface_hub-0.30.1.dist-info/top_level.txt,sha256=8KzlQJAY4miUvjAssOAJodqKOw3harNzuiwGQ9qLSSk,16
+huggingface_hub-0.30.1.dist-info/RECORD,,