github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Fix the _call of HuggingFaceHub (#16891)

Browse Source 
Fixed the following identified issue: #16849

@baskaryan
This commit is contained in:
Mohammad Mohtashim 2024-02-06 04:34:42 +05:00 committed by GitHub
parent 304f3f5fc1
commit 3c4b24b69a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/community/langchain_community/llms/huggingface_hub.py
View File

@ -127,10 +127,10 @@ class HuggingFaceHub(LLM):
response = hf("Tell me a joke.") response = hf("Tell me a joke.")
""" """
_model_kwargs = self.model_kwargs or {} _model_kwargs = self.model_kwargs or {}
params = {**_model_kwargs, **kwargs} parameters = {**_model_kwargs, **kwargs}
response = self.client.post( response = self.client.post(
json={"inputs": prompt, "params": params}, task=self.task json={"inputs": prompt, "parameters": parameters}, task=self.task
) )
response = json.loads(response.decode()) response = json.loads(response.decode())
if "error" in response: if "error" in response: