fix: :bug: fix 2 model config bugs

1 files changed, 2 insertions, 1 deletions

diff --git a/continuedev/src/continuedev/core/models.py b/continuedev/src/continuedev/core/models.py
index f03a8f29..f3bf8125 100644
--- a/continuedev/src/continuedev/core/models.py
+++ b/continuedev/src/continuedev/core/models.py
@@ -10,7 +10,7 @@ from ..libs.llm.ollama import Ollama
 from ..libs.llm.openai import OpenAI
 from ..libs.llm.replicate import ReplicateLLM
 from ..libs.llm.together import TogetherLLM
-
+from ..libs.llm.llamacpp import LlamaCpp
 
 class ContinueSDK(BaseModel):
     pass
@@ -35,6 +35,7 @@ MODEL_CLASSES = {
         AnthropicLLM,
         ReplicateLLM,
         Ollama,
+        LlamaCpp
     ]
 }