diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index a2d959a70..e31f04bb2 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -737,6 +737,7 @@ class OllamaProvider(BaseProvider, Ollama): model_id_key = "model" # TODO: add more models = [ + "gemma", "llama2", "mistral", "tinyllama",