Final_Assignment_Template

Configuration error

Coool2 commited on 14 days ago

Commit

31b8b37

1 Parent(s): 6b13586

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -200,7 +200,7 @@ def initialize_models(use_api_mode=False):
             embed_model = HuggingFaceEmbedding(
                 model_name="llamaindex/vdr-2b-multi-v1",
-                device="auto",
             trust_remote_code = True)
             # Code LLM

             embed_model = HuggingFaceEmbedding(
                 model_name="llamaindex/vdr-2b-multi-v1",
+                device="cuda",
             trust_remote_code = True)
             # Code LLM