Dinesh310 commited on
Commit
656f1b2
·
verified ·
1 Parent(s): 3b6bc0c

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="BAAI/bge-small-en-v1.5",
23
- model_kwargs={"device": "cpu", "low_cpu_mem_usage": False},
24
  encode_kwargs={"normalize_embeddings": True}
25
  )
26
  self.llm = ChatOpenAI(
 
20
  def __init__(self):
21
  self.embeddings = HuggingFaceEmbeddings(
22
  model_name="BAAI/bge-small-en-v1.5",
23
+ model_kwargs={"device": "cpu"},
24
  encode_kwargs={"normalize_embeddings": True}
25
  )
26
  self.llm = ChatOpenAI(