Adds skip test for not supported device

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Nov 5, 2024 · 480287e · 480287e
1 parent 6297694
commit 480287e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_config.py b/tests/test_config.py
@@ -154,6 +154,8 @@ def test_get_pooling_config_from_args():
     assert minilm_pooling_config.pooling_type == PoolingType.CLS.name
 
 
+@pytest.mark.skipif(current_platform.is_rocm(),
+                    reason="Xformers backend is not supported on ROCm.")
 def test_get_bert_tokenization_sentence_transformer_config():
     bge_model_config = ModelConfig(
         model="BAAI/bge-base-en-v1.5",