Dinesh310 commited on
Commit
9ee4458
·
verified ·
1 Parent(s): 4f5e730

Update src/RAG_builder.py

Browse files
Files changed (1) hide show
  1. 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="Octen/Octen-Embedding-0.6B",
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="Octen/Octen-Embedding-0.6B",
23
+ model_kwargs={"device_map": "cpu"},
24
  encode_kwargs={"normalize_embeddings": True}
25
  )
26
  self.llm = ChatOpenAI(