Coool2 commited on
Commit
4d84d69
·
1 Parent(s): ecc3536

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -3
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="onnx",
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: