Atualização do device do modelo de embedding.
Browse files
rag.py
CHANGED
@@ -67,7 +67,7 @@ class Rag:
|
|
67 |
self.embedding_model = HuggingFaceEmbeddings(
|
68 |
model_name=self.embedding_model_name,
|
69 |
multi_process=True,
|
70 |
-
model_kwargs={"device": "
|
71 |
encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
|
72 |
)
|
73 |
|
|
|
67 |
self.embedding_model = HuggingFaceEmbeddings(
|
68 |
model_name=self.embedding_model_name,
|
69 |
multi_process=True,
|
70 |
+
model_kwargs={"device": "cpu"}, # TODO: AJUSTAR DEPOIS
|
71 |
encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
|
72 |
)
|
73 |
|