From d43462a357f995d8a7b6ef0e0a2802ea40561e4f Mon Sep 17 00:00:00 2001 From: Celina Hanouti Date: Wed, 5 Feb 2025 15:26:33 +0100 Subject: [PATCH] nit --- src/huggingface_hub/inference/_providers/fal_ai.py | 2 +- src/huggingface_hub/inference/_providers/hf_inference.py | 2 +- src/huggingface_hub/inference/_providers/replicate.py | 2 +- src/huggingface_hub/inference/_providers/sambanova.py | 2 +- src/huggingface_hub/inference/_providers/together.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/huggingface_hub/inference/_providers/fal_ai.py b/src/huggingface_hub/inference/_providers/fal_ai.py index f63d50ddc7..fcb2187b1c 100644 --- a/src/huggingface_hub/inference/_providers/fal_ai.py +++ b/src/huggingface_hub/inference/_providers/fal_ai.py @@ -59,7 +59,7 @@ def prepare_request( raise ValueError( "You must provide an api_key to work with fal.ai API or log in with `huggingface-cli login`." ) - mapped_model = self.map_model(model=model) + mapped_model = self.map_model(model) headers = { **build_hf_headers(token=api_key), **headers, diff --git a/src/huggingface_hub/inference/_providers/hf_inference.py b/src/huggingface_hub/inference/_providers/hf_inference.py index ef4dd78c5f..4e7955d622 100644 --- a/src/huggingface_hub/inference/_providers/hf_inference.py +++ b/src/huggingface_hub/inference/_providers/hf_inference.py @@ -79,7 +79,7 @@ def prepare_request( ) -> RequestParameters: if extra_payload is None: extra_payload = {} - model = self.map_model(model=model) + model = self.map_model(model) url = self.build_url(model) data, json = self._prepare_payload(inputs, parameters=parameters, model=model, extra_payload=extra_payload) headers = self.prepare_headers(headers=headers, api_key=api_key) diff --git a/src/huggingface_hub/inference/_providers/replicate.py b/src/huggingface_hub/inference/_providers/replicate.py index e5284ea2d1..57d4e025f1 100644 --- a/src/huggingface_hub/inference/_providers/replicate.py +++ b/src/huggingface_hub/inference/_providers/replicate.py @@ -71,7 +71,7 @@ def prepare_request( base_url = BASE_URL logger.info("Calling Replicate provider directly.") - mapped_model = self.map_model(model=model) + mapped_model = self.map_model(model) url = _build_url(base_url, mapped_model) headers = { diff --git a/src/huggingface_hub/inference/_providers/sambanova.py b/src/huggingface_hub/inference/_providers/sambanova.py index 521e8d57a9..b212241668 100644 --- a/src/huggingface_hub/inference/_providers/sambanova.py +++ b/src/huggingface_hub/inference/_providers/sambanova.py @@ -63,7 +63,7 @@ def prepare_request( logger.info("Calling Sambanova provider directly.") headers = {**build_hf_headers(token=api_key), **headers} - mapped_model = self.map_model(model=model) + mapped_model = self.map_model(model) payload = { "messages": inputs, **{k: v for k, v in parameters.items() if v is not None}, diff --git a/src/huggingface_hub/inference/_providers/together.py b/src/huggingface_hub/inference/_providers/together.py index ac89084dd9..057190f6b8 100644 --- a/src/huggingface_hub/inference/_providers/together.py +++ b/src/huggingface_hub/inference/_providers/together.py @@ -105,7 +105,7 @@ def prepare_request( else: base_url = BASE_URL logger.info("Calling Together provider directly.") - mapped_model = self.map_model(model=model) + mapped_model = self.map_model(model) if "model" in parameters: parameters["model"] = mapped_model payload = self._prepare_payload(inputs, parameters=parameters)