Final_Assignment_Template

Configuration error

Coool2 commited on 14 days ago

Commit

4112379

1 Parent(s): 0970940

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -209,8 +209,7 @@ def initialize_models(use_api_mode=False):
             # Embedding model
             embed_model = HuggingFaceEmbedding(
                 model_name="nomic-ai/colnomic-embed-multimodal-3b",
-                device="cuda:1",
-                trust_remote_code=True)
             return proj_llm, code_llm, embed_model
         except Exception as e:

             # Embedding model
             embed_model = HuggingFaceEmbedding(
                 model_name="nomic-ai/colnomic-embed-multimodal-3b",
+                device="cuda:1")
             return proj_llm, code_llm, embed_model
         except Exception as e: