phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

support n param for hf

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-03 07:10:05 -07:00
parent 9e65d6292e
commit c7384859a3
5 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/huggingface_restapi.py
View file

@ -117,7 +117,6 @@ def completion(
inference_params = copy.deepcopy(optional_params)
inference_params.pop("details")
inference_params.pop("return_full_text")
inference_params.pop("task")
past_user_inputs = []
generated_responses = []
text = ""
@ -181,7 +180,6 @@ def completion(
inference_params = copy.deepcopy(optional_params)
inference_params.pop("details")
inference_params.pop("return_full_text")
inference_params.pop("task")
data = {
"inputs": prompt,
"parameters": inference_params,