phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

adding support for finetuned completion models

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-23 10:02:13 -07:00
parent 0634510285
commit 14ab99c3c4
3 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -248,7 +248,9 @@ def completion(
original_response=response,
additional_args={"headers": litellm.headers},
)
elif model in litellm.open_ai_text_completion_models:
elif (model in litellm.open_ai_text_completion_models or
"ft:babbage-002" in model or # support for finetuned completion models
"ft:davinci-002" in model):
openai.api_type = "openai"
openai.api_base = (
litellm.api_base