klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-08 05:39:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update default chat model to Mistral in GPT4AllProcessor config

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-11-26 03:46:01 -08:00
parent 4636390f7f
commit cc9eae5d18
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/utils/config.py
View File

@@ -80,7 +80,7 @@ class GPT4AllProcessorConfig:
class GPT4AllProcessorModel: class GPT4AllProcessorModel:
def __init__( def __init__(
self, self,
chat_model: str = "llama-2-7b-chat.ggmlv3.q4_0.bin", chat_model: str = "mistral-7b-instruct-v0.1.Q4_0.gguf",
): ):
self.chat_model = chat_model self.chat_model = chat_model
self.loaded_model = None self.loaded_model = None

3
src/khoj/utils/state.py
View File

@@ -7,6 +7,7 @@ from collections import defaultdict
# External Packages # External Packages
from pathlib import Path from pathlib import Path
from khoj.processor.embeddings import CrossEncoderModel, EmbeddingsModel from khoj.processor.embeddings import CrossEncoderModel, EmbeddingsModel
from whisper import Whisper
# Internal Packages # Internal Packages
from khoj.utils import config as utils_config from khoj.utils import config as utils_config
@@ -21,7 +22,7 @@ embeddings_model: EmbeddingsModel = None
cross_encoder_model: CrossEncoderModel = None cross_encoder_model: CrossEncoderModel = None
content_index = ContentIndex() content_index = ContentIndex()
gpt4all_processor_config: GPT4AllProcessorModel = None gpt4all_processor_config: GPT4AllProcessorModel = None
whisper_model = None whisper_model: Whisper = None
config_file: Path = None config_file: Path = None
verbose: int = 0 verbose: int = 0
host: str = None host: str = None