Add gemma

jupyterlab · Jun 20, 2024 · b5df73c · b5df73c
1 parent e28c0c0
commit b5df73c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",