huggingface-hub 0.29.1__py3-none-any.whl → 0.29.3__py3-none-any.whl

huggingface_hub/__init__.py

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

huggingface_hub/inference/_client.py

@@ -132,7 +132,7 @@ class InferenceClient:
             path will be appended to the base URL (see the [TGI Messages API](https://huggingface.co/docs/text-generation-inference/en/messages_api)
             documentation for details). When passing a URL as `model`, the client will not append any suffix path to it.
         provider (`str`, *optional*):
-            Name of the provider to use for inference. Can be `"black-forest-labs"`, `"fal-ai"`, `"fireworks-ai"`, `"hf-inference"`, `"hyperbolic"`, `"nebius"`, `"novita"`, `"replicate"`, "sambanova"` or `"together"`.
+            Name of the provider to use for inference. Can be `"black-forest-labs"`, `"cerebras"`, `"cohere"`, `"fal-ai"`, `"fireworks-ai"`, `"hf-inference"`, `"hyperbolic"`, `"nebius"`, `"novita"`, `"replicate"`, "sambanova"` or `"together"`.
             defaults to hf-inference (Hugging Face Serverless Inference API).
             If model is a URL or `base_url` is passed, then `provider` is not used.
         token (`str` or `bool`, *optional*):

huggingface_hub/inference/_generated/_async_client.py

@@ -120,7 +120,7 @@ class AsyncInferenceClient:
             path will be appended to the base URL (see the [TGI Messages API](https://huggingface.co/docs/text-generation-inference/en/messages_api)
             documentation for details). When passing a URL as `model`, the client will not append any suffix path to it.
         provider (`str`, *optional*):
-            Name of the provider to use for inference. Can be `"black-forest-labs"`, `"fal-ai"`, `"fireworks-ai"`, `"hf-inference"`, `"hyperbolic"`, `"nebius"`, `"novita"`, `"replicate"`, "sambanova"` or `"together"`.
+            Name of the provider to use for inference. Can be `"black-forest-labs"`, `"cerebras"`, `"cohere"`, `"fal-ai"`, `"fireworks-ai"`, `"hf-inference"`, `"hyperbolic"`, `"nebius"`, `"novita"`, `"replicate"`, "sambanova"` or `"together"`.
             defaults to hf-inference (Hugging Face Serverless Inference API).
             If model is a URL or `base_url` is passed, then `provider` is not used.
         token (`str` or `bool`, *optional*):

huggingface_hub/inference/_providers/__init__.py

@@ -2,6 +2,8 @@ from typing import Dict, Literal
 
 from ._common import TaskProviderHelper
 from .black_forest_labs import BlackForestLabsTextToImageTask
+from .cerebras import CerebrasConversationalTask
+from .cohere import CohereConversationalTask
 from .fal_ai import (
     FalAIAutomaticSpeechRecognitionTask,
     FalAITextToImageTask,
@@ -20,6 +22,8 @@ from .together import TogetherConversationalTask, TogetherTextGenerationTask, To
 
 PROVIDER_T = Literal[
     "black-forest-labs",
+    "cerebras",
+    "cohere",
     "fal-ai",
     "fireworks-ai",
     "hf-inference",
@@ -35,6 +39,12 @@ PROVIDERS: Dict[PROVIDER_T, Dict[str, TaskProviderHelper]] = {
     "black-forest-labs": {
         "text-to-image": BlackForestLabsTextToImageTask(),
     },
+    "cerebras": {
+        "conversational": CerebrasConversationalTask(),
+    },
+    "cohere": {
+        "conversational": CohereConversationalTask(),
+    },
     "fal-ai": {
         "automatic-speech-recognition": FalAIAutomaticSpeechRecognitionTask(),
         "text-to-image": FalAITextToImageTask(),

huggingface_hub/inference/_providers/_common.py

@@ -17,6 +17,8 @@ HARDCODED_MODEL_ID_MAPPING: Dict[str, Dict[str, str]] = {
     #
     # Example:
     # "Qwen/Qwen2.5-Coder-32B-Instruct": "Qwen2.5-Coder-32B-Instruct",
+    "cerebras": {},
+    "cohere": {},
     "fal-ai": {},
     "fireworks-ai": {},
     "hf-inference": {},

huggingface_hub/inference/_providers/cerebras.py

@@ -0,0 +1,6 @@
+from huggingface_hub.inference._providers._common import BaseConversationalTask
+
+
+class CerebrasConversationalTask(BaseConversationalTask):
+    def __init__(self):
+        super().__init__(provider="cerebras", base_url="https://api.cerebras.ai")

huggingface_hub/inference/_providers/cohere.py

@@ -0,0 +1,15 @@
+from huggingface_hub.inference._providers._common import (
+    BaseConversationalTask,
+)
+
+
+_PROVIDER = "cohere"
+_BASE_URL = "https://api.cohere.com"
+
+
+class CohereConversationalTask(BaseConversationalTask):
+    def __init__(self):
+        super().__init__(provider=_PROVIDER, base_url=_BASE_URL)
+
+    def _prepare_route(self, mapped_model: str) -> str:
+        return "/compatibility/v1/chat/completions"

huggingface_hub/inference/_providers/hf_inference.py

@@ -84,7 +84,11 @@ class HFInferenceConversational(HFInferenceTask):
         super().__init__("text-generation")
 
     def _prepare_payload_as_dict(self, inputs: Any, parameters: Dict, mapped_model: str) -> Optional[Dict]:
-        payload_model = "tgi" if mapped_model.startswith(("http://", "https://")) else mapped_model
+        payload_model = parameters.get("model") or mapped_model
+
+        if payload_model is None or payload_model.startswith(("http://", "https://")):
+            payload_model = "dummy"
+
         return {**filter_none(parameters), "model": payload_model, "messages": inputs}
 
     def _prepare_url(self, api_key: str, mapped_model: str) -> str:

huggingface_hub/utils/_subprocess.py

@@ -44,8 +44,10 @@ def capture_output() -> Generator[StringIO, None, None]:
     output = StringIO()
     previous_output = sys.stdout
     sys.stdout = output
-    yield output
-    sys.stdout = previous_output
+    try:
+        yield output
+    finally:
+        sys.stdout = previous_output
 
 
 def run_subprocess(

{huggingface_hub-0.29.1.dist-info → huggingface_hub-0.29.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface-hub
-Version: 0.29.1
+Version: 0.29.3
 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.29.1.dist-info → huggingface_hub-0.29.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-huggingface_hub/__init__.py,sha256=T-o7tRMXCYjO5nPSgmN_PAVEpFlQTOp7gh-gh8ucXak,48761
+huggingface_hub/__init__.py,sha256=-rSbs6ih-D0d2-rOjcKUbjeilJXmEIUzWlAo61vy25w,48761
 huggingface_hub/_commit_api.py,sha256=TqXmu5moVAhBa7iuyJdsqsfRTxTpGMnvsPkb4GgC3dc,32636
 huggingface_hub/_commit_scheduler.py,sha256=tfIoO1xWHjTJ6qy6VS6HIoymDycFPg0d6pBSZprrU2U,14679
 huggingface_hub/_inference_endpoints.py,sha256=SLoZOQtv_hNl0Xuafo34L--zuCZ3zSJja2tSkYkG5V4,17268
@@ -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=aiiVqLiYisaEZTOxkv90vGhsdIM-fvXdhuDwhoNbjSQ,162205
+huggingface_hub/inference/_client.py,sha256=IUpjnUpPLxHOqeAHYLPBTt5LjM18f6TEopoB_uJT3q4,162231
 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=barbsIBB5d76l3zO3Tj_2WV6Phmwfjtuq7277qHfOYg,168438
+huggingface_hub/inference/_generated/_async_client.py,sha256=IKANJspLZgHLgpmB_F-h3lfdToJkFt4V5Rc3qtJQuY4,168464
 huggingface_hub/inference/_generated/types/__init__.py,sha256=CJwdkaPbR-vzCWU1ITr4aHOHax87JaewaIs_7rKaRXE,6274
 huggingface_hub/inference/_generated/types/audio_classification.py,sha256=Jg3mzfGhCSH6CfvVvgJSiFpkz6v4nNA0G4LJXacEgNc,1573
 huggingface_hub/inference/_generated/types/audio_to_audio.py,sha256=2Ep4WkePL7oJwcp5nRJqApwviumGHbft9HhXE9XLHj4,891
@@ -77,12 +77,14 @@ huggingface_hub/inference/_generated/types/visual_question_answering.py,sha256=A
 huggingface_hub/inference/_generated/types/zero_shot_classification.py,sha256=BAiebPjsqoNa8EU35Dx0pfIv8W2c4GSl-TJckV1MaxQ,1738
 huggingface_hub/inference/_generated/types/zero_shot_image_classification.py,sha256=8J9n6VqFARkWvPfAZNWEG70AlrMGldU95EGQQwn06zI,1487
 huggingface_hub/inference/_generated/types/zero_shot_object_detection.py,sha256=GUd81LIV7oEbRWayDlAVgyLmY596r1M3AW0jXDp1yTA,1630
-huggingface_hub/inference/_providers/__init__.py,sha256=Q1hPPQgN3gKTa3NWQSANUBOB3oeCLr4miVQAVaZK8DU,5352
-huggingface_hub/inference/_providers/_common.py,sha256=8mgu95x46aRhvuHOVijczBpRJK4LFHusC_FU3t4iXGw,9200
+huggingface_hub/inference/_providers/__init__.py,sha256=654l914yPC_FZG1D-wcZzNHEn3qDDdr-gVdXf01mZ4U,5634
+huggingface_hub/inference/_providers/_common.py,sha256=soJa-5v_SIsIVeZzVBD_s7Y2puS2XGmwHn2r6-0b0cQ,9238
 huggingface_hub/inference/_providers/black_forest_labs.py,sha256=YacbRSMwTcWMCtNfLZGRnjAwyOLAM9sIj06ZUKDb7n0,2647
+huggingface_hub/inference/_providers/cerebras.py,sha256=YT1yFhXvDJiKZcqcJcA_7VZJFZVkABnv6QiEb0S90rE,246
+huggingface_hub/inference/_providers/cohere.py,sha256=zfqK2iO6VzIDbjGqi25XvD-u98pe2DwIBEDtBxdmwGA,405
 huggingface_hub/inference/_providers/fal_ai.py,sha256=pjWeMfxatAXSVJsEQf142MQVvAz5x-jtZLYXapXJFlI,3455
 huggingface_hub/inference/_providers/fireworks_ai.py,sha256=NazpDeD4agtFW6ISaXEvq5XAPVNeoG9XWk3O4NCxBNI,228
-huggingface_hub/inference/_providers/hf_inference.py,sha256=5CUR4LzPHiHfd5JN3ooP3DbOAyRgEzbQb0ZoaaiiNPY,5183
+huggingface_hub/inference/_providers/hf_inference.py,sha256=UCDiW26pbBx9P0JGnBOAoortNH2rvDURvRry2kThK6Y,5272
 huggingface_hub/inference/_providers/hyperbolic.py,sha256=qccC_gcMstGnvjmRyslgnuFVa9VAKS9w6F1ohwysvMU,1739
 huggingface_hub/inference/_providers/nebius.py,sha256=P34BO2y8MdBWqYzzt4VlkPePkXAIbMlRxvV87UhZVdU,1508
 huggingface_hub/inference/_providers/novita.py,sha256=SLOgZuAP1-Zs9NB2JmLf6kgX8R4O1Yy_64Ok9CmEZNs,745
@@ -114,7 +116,7 @@ huggingface_hub/utils/_pagination.py,sha256=hzLFLd8i_DKkPRVYzOx2CxLt5lcocEiAxDJr
 huggingface_hub/utils/_paths.py,sha256=w1ZhFmmD5ykWjp_hAvhjtOoa2ZUcOXJrF4a6O3QpAWo,5042
 huggingface_hub/utils/_runtime.py,sha256=tUyWylDgqaOXnMg39rvyusiruVN5ulcqiSwUEkQ9jjg,11195
 huggingface_hub/utils/_safetensors.py,sha256=GW3nyv7xQcuwObKYeYoT9VhURVzG1DZTbKBKho8Bbos,4458
-huggingface_hub/utils/_subprocess.py,sha256=6GpGD4qE9-Z1-Ocs3JuCLjR4NcRlknA-hAuQlqiprYY,4595
+huggingface_hub/utils/_subprocess.py,sha256=u9FFUDE7TrzQTiuEzlUnHx7S2P57GbYRV8u16GJwrFw,4625
 huggingface_hub/utils/_telemetry.py,sha256=54LXeIJU5pEGghPAh06gqNAR-UoxOjVLvKqAQscwqZs,4890
 huggingface_hub/utils/_typing.py,sha256=Dgp6TQUlpzStfVLoSvXHCBP4b3NzHZ8E0Gg9mYAoDS4,2903
 huggingface_hub/utils/_validators.py,sha256=dDsVG31iooTYrIyi5Vwr1DukL0fEmJwu3ceVNduhsuE,9204
@@ -123,9 +125,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=ZgdphuTnwAIaUKnnD2P7qVvNHpzHAyrYoItkiV0aEjQ,9835
-huggingface_hub-0.29.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-huggingface_hub-0.29.1.dist-info/METADATA,sha256=B8dl2q55ILPp7jxGZ3Nx0zT0AlCd74Z0ipYxygbW3FI,13480
-huggingface_hub-0.29.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-huggingface_hub-0.29.1.dist-info/entry_points.txt,sha256=Y3Z2L02rBG7va_iE6RPXolIgwOdwUFONyRN3kXMxZ0g,131
-huggingface_hub-0.29.1.dist-info/top_level.txt,sha256=8KzlQJAY4miUvjAssOAJodqKOw3harNzuiwGQ9qLSSk,16
-huggingface_hub-0.29.1.dist-info/RECORD,,
+huggingface_hub-0.29.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+huggingface_hub-0.29.3.dist-info/METADATA,sha256=2YNFg7jwDyl9lQIFl4PZtBLsxPyi9VXExk98eKV0jKo,13480
+huggingface_hub-0.29.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+huggingface_hub-0.29.3.dist-info/entry_points.txt,sha256=Y3Z2L02rBG7va_iE6RPXolIgwOdwUFONyRN3kXMxZ0g,131
+huggingface_hub-0.29.3.dist-info/top_level.txt,sha256=8KzlQJAY4miUvjAssOAJodqKOw3harNzuiwGQ9qLSSk,16
+huggingface_hub-0.29.3.dist-info/RECORD,,