fix embedding model name argument

autogluon · Jul 31, 2024 · f564c03 · f564c03
1 parent 6972c1c
commit f564c03
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/agrag/agrag.py b/src/agrag/agrag.py
@@ -186,7 +186,7 @@ def initialize_data_module(self):
     def initialize_embeddings_module(self):
         """Initializes the Embedding module."""
         self.embedding_module = EmbeddingModule(
-            model_name=self.args.hf_embedding_model,
+            model_name=self.args.embedding_model,
             pooling_strategy=self.args.pooling_strategy,
             normalize_embeddings=self.args.normalize_embeddings,
             hf_model_params=self.args.hf_model_params,

diff --git a/src/agrag/args.py b/src/agrag/args.py
@@ -171,13 +171,13 @@ def data_file_extns(self, value):
         self.config["data"]["file_extns"] = value
 
     @property
-    def hf_embedding_model(self):
+    def embedding_model(self):
         return self.config.get("embedding", {}).get(
             "embedding_model", self.embedding_defaults.get("DEFAULT_EMBEDDING_MODEL")
         )
 
-    @hf_embedding_model.setter
-    def hf_embedding_model(self, value):
+    @embedding_model.setter
+    def embedding_model(self, value):
         self.config["embedding"]["embedding_model"] = value
 
     @property