Dinesh310 commited on
Commit
3b6bc0c
·
verified ·
1 Parent(s): 383654f

Update src/RAG_builder.py

Browse files
Files changed (1) hide show
  1. src/RAG_builder.py +1 -0
src/RAG_builder.py CHANGED
@@ -20,6 +20,7 @@ class ProjectRAGGraph:
20
  def __init__(self):
21
  self.embeddings = HuggingFaceEmbeddings(
22
  model_name="BAAI/bge-small-en-v1.5",
 
23
  encode_kwargs={"normalize_embeddings": True}
24
  )
25
  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", "low_cpu_mem_usage": False},
24
  encode_kwargs={"normalize_embeddings": True}
25
  )
26
  self.llm = ChatOpenAI(