add: items in known_models

luochen1990 · Aug 14, 2024 · b109233 · b109233
1 parent dd9962f
commit b109233
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/ai_powered/llm/known_models.py b/src/ai_powered/llm/known_models.py
@@ -72,8 +72,10 @@ def equals(s: str) -> Callable[[str], bool]:
         known_model_list = [
             KnownModel("gorilla-llm/gorilla-openfunctions-v2-gguf/gorilla-openfunctions-v2-q4_K_M.gguf", ALL_FEATURES),
             KnownModel("lmstudio-community/Meta-Llama-3-8B-Instruct-GGUF/Meta-Llama-3-8B-Instruct-Q4_K_M.gguf", set()),
-            KnownModel("deepseek-coder-v2", set()),
             KnownModel("lmstudio-community/DeepSeek-Coder-V2-Lite-Instruct-GGUF", set()),
+            KnownModel("firefunction-v2:70b-q2_K", {ModelFeature.tools}),
+            KnownModel("deepseek-coder-v2", set()),
+            KnownModel("llama3.1", set()),
         ]
     ),
 ]