fix: Added language to KFST

KennethEnevoldsen · Jul 23, 2024 · c89856a · c89856a
1 parent 07a7ec3
commit c89856a
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),