Coool2 commited on
Commit
999040b
·
1 Parent(s): ca4a1b3

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
agent.py CHANGED
@@ -197,8 +197,9 @@ def initialize_models(use_api_mode=False):
197
 
198
 
199
  embed_model = HuggingFaceEmbedding(
200
- model_name="nomic-ai/nomic-embed-multimodal-3b",
201
- device="cuda:1")
 
202
  proj_llm = QwenVL7BCustomLLM()
203
 
204
  # Code LLM
 
197
 
198
 
199
  embed_model = HuggingFaceEmbedding(
200
+ model_name="llamaindex/vdr-2b-multi-v1",
201
+ device="cuda:1",
202
+ trust_remote_code = True)
203
  proj_llm = QwenVL7BCustomLLM()
204
 
205
  # Code LLM