diff --git a/run_localGPT.py b/run_localGPT.py index b6450e02..a553825d 100644 --- a/run_localGPT.py +++ b/run_localGPT.py @@ -26,6 +26,7 @@ ) from constants import ( + CHROMA_SETTINGS, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY, MODEL_ID, @@ -122,6 +123,7 @@ def retrieval_qa_pipline(device_type, use_history, promptTemplate_type="llama"): db = Chroma( persist_directory=PERSIST_DIRECTORY, embedding_function=embeddings, + client_settings=CHROMA_SETTINGS, ) retriever = db.as_retriever()