From 6297694a7dc646e272c9f752c90cbc4d8dbbb36e Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Tue, 5 Nov 2024 15:34:23 -0300 Subject: [PATCH] Fix yapf lint Signed-off-by: Flavia Beo --- tests/model_executor/test_model_load_with_params.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/model_executor/test_model_load_with_params.py b/tests/model_executor/test_model_load_with_params.py index d9d10c1f4238e..0517cc397602c 100644 --- a/tests/model_executor/test_model_load_with_params.py +++ b/tests/model_executor/test_model_load_with_params.py @@ -10,6 +10,7 @@ MODEL_NAME = os.environ.get("MODEL_NAME", "BAAI/bge-base-en-v1.5") REVISION = os.environ.get("REVISION", "main") + @pytest.mark.skipif(current_platform.is_rocm(), reason="Xformers backend is not supported on ROCm.") def test_model_loading_with_params(vllm_runner):