diff options
Diffstat (limited to 'server/continuedev/core/config.py')
-rw-r--r-- | server/continuedev/core/config.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/server/continuedev/core/config.py b/server/continuedev/core/config.py index 2bbb42cc..bf555b59 100644 --- a/server/continuedev/core/config.py +++ b/server/continuedev/core/config.py @@ -2,7 +2,7 @@ from typing import Dict, List, Optional, Type from pydantic import BaseModel, Field, validator -from ..libs.llm.openai_free_trial import OpenAIFreeTrial +from ..libs.llm import Ollama from .context import ContextProvider from .main import Policy, Step from .models import Models @@ -48,8 +48,14 @@ class ContinueConfig(BaseModel): ) models: Models = Field( Models( - default=OpenAIFreeTrial(model="gpt-4"), - summarize=OpenAIFreeTrial(model="gpt-3.5-turbo"), + default=Ollama( + title="CodeLlama-7b-Instruct", + model="codellama:7b-instruct" + ), + summarize=Ollama( + title="CodeLlama-7b-Instruct", + model="codellama:7b-instruct" + ) ), description="Configuration for the models used by Continue. Read more about how to configure models in the documentation.", ) |