Update embedding_model.py

embedding_model.py

@@ -72,7 +72,7 @@ class EmbeddingModel(models.Transformer):
             json.dump(self.get_config_dict(), f, indent=4)
 
     @staticmethod
-    def load(load_dir: str, **kwargs) -> "EmbeddingModel":
+    def load(load_dir: str, *args, **kwargs) -> "EmbeddingModel":
         with open(os.path.join(load_dir, "config.json")) as fIn:
             config = json.load(fIn)
-        return EmbeddingModel(**config, **kwargs)
+        return EmbeddingModel(**config, *args, **kwargs)