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

petals on main

Browse source
This commit is contained in:
ishaan-jaff 2023-09-19 09:27:28 -07:00
parent 9c37619e5c
commit 5f808f9f87
1 changed files with 0 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -956,12 +956,9 @@ def completion(
response = model_response response = model_response
elif ( elif (
custom_llm_provider == "petals" custom_llm_provider == "petals"
or custom_llm_provider == "petals-team"
or model in litellm.petals_models or model in litellm.petals_models
): ):
custom_llm_provider = "petals" custom_llm_provider = "petals"
print("model on petals")
print(model)
model_response = petals.completion( model_response = petals.completion(
model=model, model=model,
messages=messages, messages=messages,