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

adding support for finetuned completion models

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-23 10:02:13 -07:00
parent f3cf199cde
commit 228669caa1
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