From c89856a5be9381e3d50b3c84eb7b7e17901c2c45 Mon Sep 17 00:00:00 2001 From: Kenneth Enevoldsen Date: Tue, 23 Jul 2024 10:21:46 +0200 Subject: [PATCH] fix: Added language to KFST --- src/seb/registered_models/sentence_transformer_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/seb/registered_models/sentence_transformer_models.py b/src/seb/registered_models/sentence_transformer_models.py index 7a43ce7..d508fbe 100644 --- 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),