Skip to content

Commit

Permalink
Added Memo-bert-03 model
Browse files Browse the repository at this point in the history
  • Loading branch information
x-tabdeveloping committed Aug 23, 2024
1 parent 6eb88ec commit 5c60d65
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions src/seb/registered_models/sentence_transformer_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -524,6 +524,25 @@ def create_xlmroberta_en_da_sv_nb() -> SebModel:
)


@models.register("MeMo-BERT-03")
def create_memo_bert_03() -> SebModel:
hf_name = "MiMe-MeMo/MeMo-BERT-03"
meta = ModelMeta(
name=hf_name.split("/")[-1],
huggingface_name=hf_name,
reference=f"https://huggingface.co/{hf_name}",
open_source=True,
languages=["da"],
embedding_size=768,
architecture="BERT",
release_date=date(2023, 7, 31),
)
return SebModel(
encoder=LazyLoadEncoder(partial(wrap_sentence_transformer, model_name=hf_name)), # type: ignore
meta=meta,
)


if __name__ == "__main__":
import seb

Expand Down

0 comments on commit 5c60d65

Please sign in to comment.