Use larger LLM and embedding models

nirogu · Sep 13, 2024 · f6fbcb5 · f6fbcb5
1 parent c8438d2
commit f6fbcb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"