Update embedding_model.py
Browse files- embedding_model.py +1 -5
embedding_model.py
CHANGED
|
@@ -75,8 +75,4 @@ class EmbeddingModel(models.Transformer):
|
|
| 75 |
def load(load_dir: str, **kwargs) -> "EmbeddingModel":
|
| 76 |
with open(os.path.join(load_dir, "config.json")) as fIn:
|
| 77 |
config = json.load(fIn)
|
| 78 |
-
return EmbeddingModel(
|
| 79 |
-
**config,
|
| 80 |
-
model_args={"trust_remote_code": True},
|
| 81 |
-
tokenizer_args={"trust_remote_code": True}
|
| 82 |
-
)
|
|
|
|
| 75 |
def load(load_dir: str, **kwargs) -> "EmbeddingModel":
|
| 76 |
with open(os.path.join(load_dir, "config.json")) as fIn:
|
| 77 |
config = json.load(fIn)
|
| 78 |
+
return EmbeddingModel(**config, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|