Update src/RAG_builder.py
Browse files- src/RAG_builder.py +1 -1
src/RAG_builder.py
CHANGED
|
@@ -20,7 +20,7 @@ class ProjectRAGGraph:
|
|
| 20 |
def __init__(self):
|
| 21 |
self.embeddings = HuggingFaceEmbeddings(
|
| 22 |
model_name="BAAI/bge-small-en-v1.5",
|
| 23 |
-
model_kwargs={"device": "cpu"
|
| 24 |
encode_kwargs={"normalize_embeddings": True}
|
| 25 |
)
|
| 26 |
self.llm = ChatOpenAI(
|
|
|
|
| 20 |
def __init__(self):
|
| 21 |
self.embeddings = HuggingFaceEmbeddings(
|
| 22 |
model_name="BAAI/bge-small-en-v1.5",
|
| 23 |
+
model_kwargs={"device": "cpu"},
|
| 24 |
encode_kwargs={"normalize_embeddings": True}
|
| 25 |
)
|
| 26 |
self.llm = ChatOpenAI(
|