Spaces:
Configuration error
Configuration error
Update agent.py
Browse files
agent.py
CHANGED
@@ -197,8 +197,9 @@ def initialize_models(use_api_mode=False):
|
|
197 |
|
198 |
|
199 |
embed_model = HuggingFaceEmbedding(
|
200 |
-
model_name="
|
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
|