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

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
elif (
custom_llm_provider == "petals"
or custom_llm_provider == "petals-team"
or model in litellm.petals_models
):
custom_llm_provider = "petals"
print("model on petals")
print(model)
model_response = petals.completion(
model=model,
messages=messages,