Spaces:
Configuration error
Configuration error
Update agent.py
Browse files
agent.py
CHANGED
@@ -211,9 +211,7 @@ def initialize_models(use_api_mode=False):
|
|
211 |
model_name="llamaindex/vdr-2b-multi-v1",
|
212 |
device="cpu",
|
213 |
trust_remote_code=True,
|
214 |
-
backend="
|
215 |
-
model_kwargs={"provider": "CPUExecutionProvider"}
|
216 |
-
)
|
217 |
|
218 |
return proj_llm, code_llm, embed_model
|
219 |
except Exception as e:
|
|
|
211 |
model_name="llamaindex/vdr-2b-multi-v1",
|
212 |
device="cpu",
|
213 |
trust_remote_code=True,
|
214 |
+
backend="openvino")
|
|
|
|
|
215 |
|
216 |
return proj_llm, code_llm, embed_model
|
217 |
except Exception as e:
|