Update agent.py
Browse files
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="
|
| 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 |
)
|