Merge pull request #182 from KennethEnevoldsen/add_kfst_model

fix: Added language to KFST
KennethEnevoldsen · Jul 23, 2024 · 324dde4 · 324dde4
2 parents e8848e2 + c89856a
commit 324dde4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/seb/registered_models/sentence_transformer_models.py b/src/seb/registered_models/sentence_transformer_models.py
@@ -513,6 +513,7 @@ def create_xlmroberta_en_da_sv_nb() -> SebModel:
         huggingface_name=hf_name,
         reference=f"https://huggingface.co/{hf_name}",
         open_source=True,
+        languages=["da", "nb", "sv", "en"],
         embedding_size=768,
         architecture="XLM-R",
         release_date=date(2024, 2, 22),