Update src/RAG_builder.py
Browse files- src/RAG_builder.py +0 -1
src/RAG_builder.py
CHANGED
|
@@ -20,7 +20,6 @@ class ProjectRAGGraph:
|
|
| 20 |
def __init__(self):
|
| 21 |
self.embeddings = HuggingFaceEmbeddings(
|
| 22 |
model_name="Octen/Octen-Embedding-0.6B",
|
| 23 |
-
model_kwargs={"device_map": "cpu"},
|
| 24 |
encode_kwargs={"normalize_embeddings": True}
|
| 25 |
)
|
| 26 |
self.llm = ChatOpenAI(
|
|
|
|
| 20 |
def __init__(self):
|
| 21 |
self.embeddings = HuggingFaceEmbeddings(
|
| 22 |
model_name="Octen/Octen-Embedding-0.6B",
|
|
|
|
| 23 |
encode_kwargs={"normalize_embeddings": True}
|
| 24 |
)
|
| 25 |
self.llm = ChatOpenAI(
|