From 181b6a7dd2de046302f6c3bcb7e009eaed13dc1a Mon Sep 17 00:00:00 2001 From: Celina Hanouti Date: Wed, 5 Feb 2025 15:34:01 +0100 Subject: [PATCH] fix hf-inference --- .../inference/_providers/hf_inference.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/src/huggingface_hub/inference/_providers/hf_inference.py b/src/huggingface_hub/inference/_providers/hf_inference.py index 91f7014df4..cffb2c28df 100644 --- a/src/huggingface_hub/inference/_providers/hf_inference.py +++ b/src/huggingface_hub/inference/_providers/hf_inference.py @@ -2,12 +2,7 @@ from typing import Any, Dict, List, Optional, Tuple, Union from huggingface_hub.constants import ENDPOINT -from huggingface_hub.inference._common import ( - RequestParameters, - TaskProviderHelper, - _b64_encode, - _open_as_binary, -) +from huggingface_hub.inference._common import RequestParameters, TaskProviderHelper, _b64_encode, _open_as_binary from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status @@ -81,9 +76,7 @@ def prepare_request( extra_payload = {} mapped_model = self.map_model(model) url = self.build_url(mapped_model) - data, json = self._prepare_payload( - inputs, parameters=parameters, model=mapped_model, extra_payload=extra_payload - ) + data, json = self._prepare_payload(inputs, parameters=parameters, model=model, extra_payload=extra_payload) headers = self.prepare_headers(headers=headers, api_key=api_key) return RequestParameters(