forgot, code still trash tho

DuckyBlender · Jul 12, 2024 · a02f963 · a02f963
1 parent 258d38d
commit a02f963
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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,
         ]
     }