Skip to content

Commit

Permalink
fix: remove duplicate e5 mistral model
Browse files Browse the repository at this point in the history
  • Loading branch information
Your Name committed Jan 29, 2024
1 parent edff86b commit 2de6ee2
Showing 1 changed file with 0 additions and 17 deletions.
17 changes: 0 additions & 17 deletions src/seb/registered_models/e5_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,20 +144,3 @@ def create_multilingual_e5_large() -> EmbeddingModel:
loader=partial(E5Wrapper, model_name=hf_name), # type: ignore
meta=meta,
)


@models.register("intfloat/e5-mistral-7b-instruct")
def create_multilingual_e5_mistral_7b_instruct() -> EmbeddingModel:
hf_name = "intfloat/e5-mistral-7b-instruct"
meta = ModelMeta(
name=hf_name.split("/")[-1],
huggingface_name=hf_name,
reference=f"https://huggingface.co/{hf_name}",
languages=[],
open_source=True,
embedding_size=4096,
)
return EmbeddingModel(
loader=partial(E5Wrapper, model_name=hf_name), # type: ignore
meta=meta,
)

0 comments on commit 2de6ee2

Please sign in to comment.