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

adding first-party + custom prompt templates for huggingface

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-04 14:48:16 -07:00
parent a474b89779
commit 2384806cfd
10 changed files with 186 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -563,8 +563,8 @@ def completion(
logger_fn=logger_fn,
encoding=encoding,
api_key=huggingface_key,
logging_obj=logging
logging_obj=logging,
custom_prompt_dict=litellm.custom_prompt_dict
)
if "stream" in optional_params and optional_params["stream"] == True:
# don't try to access stream object,