From b109233248992949b0521199e8f033fd183bf65d Mon Sep 17 00:00:00 2001 From: LuoChen Date: Wed, 14 Aug 2024 21:12:39 +0800 Subject: [PATCH] add: items in known_models --- src/ai_powered/llm/known_models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/ai_powered/llm/known_models.py b/src/ai_powered/llm/known_models.py index cc13de4..262e641 100644 --- 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()), ] ), ]