Added Ollama LLMs for LLaMa and Mistral

This commit is contained in:
Kade Heckel 2024-05-09 12:06:51 +01:00
parent dde2e3249a
commit 256107c999
2 changed files with 19 additions and 1 deletions

View file

@ -6620,7 +6620,7 @@ def get_max_tokens(model: str):
raise Exception()
except:
raise Exception(
"This model isn't mapped yet. Add it here - https://github.com/BerriAI/litellm/blob/main/model_prices_and_context_window.json"
f"Model {model} from provider {custom_llm_provider} isn't mapped yet. Add it here - https://github.com/BerriAI/litellm/blob/main/model_prices_and_context_window.json"
)