Update agent.py

agent.py

@@ -209,12 +209,9 @@ def initialize_models(use_api_mode=False):
             # Embedding model
             embed_model = HuggingFaceEmbedding(
                 model_name="llamaindex/vdr-2b-multi-v1",
-                device="cpu",
+                device="cuda",
                 trust_remote_code=True,
-                model_kwargs={
-                    "torch_dtype": "auto",
-                    "low_cpu_mem_usage": True
-                }
+                model_kwargs={"torch_dtype": "auto"},
             )
     
             return proj_llm, code_llm, embed_model