Update embedding_model.py

embedding_model.py

@@ -10,9 +10,8 @@ MODEL_PATH_IN_MODULE = "0_Model"
 
 
 class EmbeddingModel(models.Transformer):
-    def __init__(self, model_name_or_path: str, *args, **kwargs):
-        self.model_name_or_path = model_name_or_path
-        super().__init__(model_name_or_path, *args, **kwargs)
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
 
     def tokenize(
         self,
@@ -72,7 +71,7 @@ class EmbeddingModel(models.Transformer):
             json.dump(self.get_config_dict(), f, indent=4)
 
     @staticmethod
-    def load(load_dir: str, *args, **kwargs) -> "EmbeddingModel":
+    def load(load_dir: str, **kwargs) -> "EmbeddingModel":
         with open(os.path.join(load_dir, "config.json")) as fIn:
             config = json.load(fIn)
-        return EmbeddingModel(**config, *args, **kwargs)
+        return EmbeddingModel(**config, **kwargs)