fixes

Author: hanouticelina

src/huggingface_hub/inference/_providers/__init__.py

@@ -27,7 +27,7 @@
     NebiusTextToImageTask,
 )
 from .novita import NovitaConversationalTask, NovitaTextGenerationTask, NovitaTextToVideoTask
-from .nscale import NscaleChatCompletion, NscaleTextToImageTask
+from .nscale import NscaleConversationalTask, NscaleTextToImageTask
 from .openai import OpenAIConversationalTask
 from .replicate import ReplicateTask, ReplicateTextToImageTask, ReplicateTextToSpeechTask
 from .sambanova import SambanovaConversationalTask, SambanovaFeatureExtractionTask

src/huggingface_hub/inference/_providers/nscale.py

@@ -1,27 +1,28 @@
 import base64
 from typing import Any, Dict, Optional, Union
 
+from huggingface_hub.hf_api import InferenceProviderMapping
 from huggingface_hub.inference._common import RequestParameters, _as_dict
 
-from ._common import (
-    BaseConversationalTask,
-    TaskProviderHelper,
-    filter_none,
-)
+from ._common import BaseConversationalTask, TaskProviderHelper, filter_none
 
 
 class NscaleConversationalTask(BaseConversationalTask):
     def __init__(self):
         super().__init__(provider="nscale", base_url="https://inference.api.nscale.com")
-        
+
+
 class NscaleTextToImageTask(TaskProviderHelper):
     def __init__(self):
         super().__init__(provider="nscale", base_url="https://inference.api.nscale.com", task="text-to-image")
+
     def _prepare_route(self, mapped_model: str, api_key: str) -> str:
         return "/v1/images/generations"
 
-    def _prepare_payload_as_dict(self, inputs: Any, parameters: Dict, provider_mapping_info: InferenceProviderMapping) -> Optional[Dict]:
-    mapped_model = provider_mapping_info.provider_id
+    def _prepare_payload_as_dict(
+        self, inputs: Any, parameters: Dict, provider_mapping_info: InferenceProviderMapping
+    ) -> Optional[Dict]:
+        mapped_model = provider_mapping_info.provider_id
         # Combine all parameters except inputs and parameters
         parameters = filter_none(parameters)
         if "width" in parameters and "height" in parameters:

tests/test_inference_providers.py

@@ -33,7 +33,7 @@
 from huggingface_hub.inference._providers.hyperbolic import HyperbolicTextGenerationTask, HyperbolicTextToImageTask
 from huggingface_hub.inference._providers.nebius import NebiusFeatureExtractionTask, NebiusTextToImageTask
 from huggingface_hub.inference._providers.novita import NovitaConversationalTask, NovitaTextGenerationTask
-from huggingface_hub.inference._providers.nscale import NscaleChatCompletion, NscaleTextToImageTask
+from huggingface_hub.inference._providers.nscale import NscaleConversationalTask, NscaleTextToImageTask
 from huggingface_hub.inference._providers.openai import OpenAIConversationalTask
 from huggingface_hub.inference._providers.replicate import ReplicateTask, ReplicateTextToSpeechTask
 from huggingface_hub.inference._providers.sambanova import SambanovaConversationalTask, SambanovaFeatureExtractionTask
@@ -858,7 +858,7 @@ def test_prepare_route_text_to_image(self):
         assert helper._prepare_route("model_name", "api_key") == "/v1/images/generations"
 
     def test_prepare_route_chat_completion(self):
-        helper = NscaleChatCompletion()
+        helper = NscaleConversationalTask()
         assert helper._prepare_route("model_name", "api_key") == "/v1/chat/completions"
 
     def test_prepare_payload_with_size_conversion(self):
@@ -869,7 +869,12 @@ def test_prepare_payload_with_size_conversion(self):
                 "width": 512,
                 "height": 512,
             },
-            "stabilityai/stable-diffusion-xl-base-1.0",
+            InferenceProviderMapping(
+                hf_model_id="stabilityai/stable-diffusion-xl-base-1.0",
+                providerId="stabilityai/stable-diffusion-xl-base-1.0",
+                task="text-to-image",
+                status="live",
+            ),
         )
         assert payload == {
             "prompt": "a beautiful landscape",
@@ -888,7 +893,12 @@ def test_prepare_payload_as_dict(self):
                 "cfg_scale": 7.5,
                 "num_inference_steps": 50,
             },
-            "stabilityai/stable-diffusion-xl-base-1.0",
+            InferenceProviderMapping(
+                hf_model_id="stabilityai/stable-diffusion-xl-base-1.0",
+                providerId="stabilityai/stable-diffusion-xl-base-1.0",
+                task="text-to-image",
+                status="live",
+            ),
         )
         assert "width" not in payload
         assert "height" not in payload