iasbeck commited on
Commit
121fbd1
1 Parent(s): a21dabb

Atualização do device do modelo de embedding.

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
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": "cuda"}, # TODO: AJUSTAR DEPOIS
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