From 5f808f9f87809ec62e38a15d44642a6ec739e79c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 19 Sep 2023 09:27:28 -0700 Subject: [PATCH] petals on main --- litellm/main.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 28292a6c4..c13183897 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -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,