Replace bakllava by moondream

moondream is a much smaller vision model and perfectly suitable for the tests we want to run.
matlab-deep-learning · Aug 2, 2024 · 172a4da · 172a4da
1 parent ea8874b
commit 172a4da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -30,7 +30,7 @@ jobs:
       - name: Pull models
         run: |
             ollama pull mistral
-            ollama pull bakllava
+            ollama pull moondream
             OLLAMA_HOST=127.0.0.1:11435 ollama pull qwen2:0.5b
       - name: Set up MATLAB
         uses: matlab-actions/setup-matlab@v2

diff --git a/tests/tollamaChat.m b/tests/tollamaChat.m
@@ -99,7 +99,7 @@ function seedFixesResult(testCase)
         end
 
         function generateWithImages(testCase)
-            chat = ollamaChat("bakllava");
+            chat = ollamaChat("moondream");
             image_path = "peppers.png";
             emptyMessages = messageHistory;
             messages = addUserMessageWithImages(emptyMessages,"What is in the image?",image_path);