Coool2 commited on
Commit
589ecc5
·
verified ·
1 Parent(s): 1bd5d26

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -1
agent.py CHANGED
@@ -97,7 +97,11 @@ code_llm = HuggingFaceLLM(
97
  generate_kwargs={"temperature": 0.0, "do_sample": False}
98
  )
99
 
100
- embed_model = HuggingFaceEmbedding("BAAI/bge-visualized-m3")
 
 
 
 
101
 
102
  Settings.llm = proj_llm
103
  Settings.embed_model = embed_model
 
97
  generate_kwargs={"temperature": 0.0, "do_sample": False}
98
  )
99
 
100
+ model = HuggingFaceEmbedding(
101
+ model_name="llamaindex/vdr-2b-multi-v1",
102
+ device="cpu", # "mps" for mac, "cuda" for nvidia GPUs
103
+ trust_remote_code=True,
104
+ )
105
 
106
  Settings.llm = proj_llm
107
  Settings.embed_model = embed_model