Coool2 commited on
Commit
92ae224
·
1 Parent(s): 59f49c1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -0
agent.py CHANGED
@@ -209,6 +209,7 @@ def initialize_models(use_api_mode=False):
209
  # Embedding model
210
  embed_model = HuggingFaceEmbedding(
211
  model_name="llamaindex/vdr-2b-multi-v1",
 
212
  trust_remote_code=True,
213
  backend="onnx",
214
  model_kwargs={"provider": "CPUExecutionProvider"}
 
209
  # Embedding model
210
  embed_model = HuggingFaceEmbedding(
211
  model_name="llamaindex/vdr-2b-multi-v1",
212
+ device="cpu",
213
  trust_remote_code=True,
214
  backend="onnx",
215
  model_kwargs={"provider": "CPUExecutionProvider"}