huggingface-hub 0.29.1__tar.gz → 0.29.3rc0__tar.gz

{huggingface_hub-0.29.1 → huggingface_hub-0.29.3rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface_hub
-Version: 0.29.1
+Version: 0.29.3rc0
 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 → huggingface_hub-0.29.3rc0}/src/huggingface_hub/__init__.py

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

{huggingface_hub-0.29.1 → huggingface_hub-0.29.3rc0}/src/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-0.29.1 → huggingface_hub-0.29.3rc0}/src/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-0.29.1 → huggingface_hub-0.29.3rc0}/src/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-0.29.1 → huggingface_hub-0.29.3rc0}/src/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-0.29.3rc0/src/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-0.29.3rc0/src/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-0.29.1 → huggingface_hub-0.29.3rc0}/src/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-0.29.1 → huggingface_hub-0.29.3rc0}/src/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 → huggingface_hub-0.29.3rc0}/src/huggingface_hub.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface-hub
-Version: 0.29.1
+Version: 0.29.3rc0
 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 → huggingface_hub-0.29.3rc0}/src/huggingface_hub.egg-info/SOURCES.txt

@@ -91,6 +91,8 @@ src/huggingface_hub/inference/_generated/types/zero_shot_object_detection.py
 src/huggingface_hub/inference/_providers/__init__.py
 src/huggingface_hub/inference/_providers/_common.py
 src/huggingface_hub/inference/_providers/black_forest_labs.py
+src/huggingface_hub/inference/_providers/cerebras.py
+src/huggingface_hub/inference/_providers/cohere.py
 src/huggingface_hub/inference/_providers/fal_ai.py
 src/huggingface_hub/inference/_providers/fireworks_ai.py
 src/huggingface_hub/inference/_providers/hf_inference.py