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

Update src/RAG_builder.py

Browse files
Files changed (1) hide show
  1. 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(