From f6fbcb5d220bea977a94cda0d76baa3fb6b7dddd Mon Sep 17 00:00:00 2001 From: nirogu Date: Fri, 13 Sep 2024 15:28:25 -0500 Subject: [PATCH] Use larger LLM and embedding models --- config.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/config.yaml b/config.yaml index 45999e3..fd9b1d7 100644 --- a/config.yaml +++ b/config.yaml @@ -1,7 +1,7 @@ index_directory: "./rag_index" chunk_size: 1024 -embedding_model: "thenlper/gte-small" -ollama_model: "tinyllama" +embedding_model: "thenlper/gte-large" +ollama_model: "mixtral" chroma_collection: "sagemaker_docs" documents_dir: "sagemaker_documentation" query_history: "queries_database.jsonl"