From a02f96355b8cd84f27e23658b222259fe7d72f11 Mon Sep 17 00:00:00 2001
From: Alan Klas <alan@alanklas.com>
Date: Fri, 12 Jul 2024 23:55:24 +0200
Subject: [PATCH] forgot, code still trash tho

---
 src/models.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/models.rs b/src/models.rs
index a901924..54f1b92 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -66,7 +66,9 @@ impl ModelType {
             ModelType::Phi3Llava,
             ModelType::Moondream,
             ModelType::TinyStories,
-            ModelType::Caveman
+            ModelType::Caveman,
+            ModelType::Gemma2,
+            ModelType::Smegmma,
         ]
     }