Add gemma

jupyterlab · Feb 21, 2024 · d862068 · d862068
1 parent 5dd82e7
commit d862068
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
@@ -638,6 +638,7 @@ class OllamaProvider(BaseProvider, Ollama):
     model_id_key = "model"
     # TODO: add more
     models = [
+        "gemma",
         "llama2",
         "mistral",
         "tinyllama",