[Bugfix] Fix RobertaModel loading (#11940)

Signed-off-by: NickLucche <nlucches@redhat.com>
2025-01-11 15:05:09 +01:00
parent a991f7d508
commit d697dc01b4
3 changed files with 67 additions and 12 deletions
--- a/tests/models/embedding/language/test_embedding.py
+++ b/tests/models/embedding/language/test_embedding.py
@@ -25,6 +25,7 @@ from ..utils import check_embeddings_close
        pytest.param("ssmits/Qwen2-7B-Instruct-embed-base"),
        pytest.param("Alibaba-NLP/gte-Qwen2-1.5B-instruct"),
        pytest.param("Alibaba-NLP/gte-Qwen2-7B-instruct"),
+        pytest.param("sentence-transformers/stsb-roberta-base-v2"),
    ],
 )
@pytest.mark.parametrize("dtype", ["half"])