diff --git a/litellm/__init__.py b/litellm/__init__.py index 1de60c761..510832a57 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -335,9 +335,9 @@ baseten_models: List = [ # used for token counting # Azure returns gpt-35-turbo in their responses, we need to map this to azure/gpt-3.5-turbo for token counting azure_llms = { - "gpt-35-turbo": "azure/gpt-3.5-turbo", - "gpt-35-turbo-16k": "azure/gpt-3.5-turbo-16k", - "gpt-35-turbo-instruct": "azure/gpt-3.5-turbo-instruct", + "gpt-35-turbo": "azure/gpt-35-turbo", + "gpt-35-turbo-16k": "azure/gpt-35-turbo-16k", + "gpt-35-turbo-instruct": "azure/gpt-35-turbo-instruct", } petals_models = [