vijusudhi commited on
Commit
94e488a
·
verified ·
1 Parent(s): 1835a4f

Update embedding_model.py

Browse files
Files changed (1) hide show
  1. 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)