Coool2 commited on
Commit
687967c
·
verified ·
1 Parent(s): 1be3511

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -87,7 +87,7 @@ code_llm = HuggingFaceLLM(
87
 
88
  embed_model = HuggingFaceEmbedding(
89
  model_name="llamaindex/vdr-2b-multi-v1",
90
- device="cuda",
91
  trust_remote_code=True,
92
  model_kwargs={"torch_dtype": "auto"}
93
  )
 
87
 
88
  embed_model = HuggingFaceEmbedding(
89
  model_name="llamaindex/vdr-2b-multi-v1",
90
+ device="cpu",
91
  trust_remote_code=True,
92
  model_kwargs={"torch_dtype": "auto"}
93
  )