From 37b6cb49852369945b4560efc8955a992f9b12cb Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Tue, 25 Feb 2025 18:01:15 +0800 Subject: [PATCH] [CI/Build] Fix V1 LoRA failure (#13767) --- tests/lora/test_gemma.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/lora/test_gemma.py b/tests/lora/test_gemma.py index a1b4c897c45ef..bbdfbe37175e1 100644 --- a/tests/lora/test_gemma.py +++ b/tests/lora/test_gemma.py @@ -41,6 +41,8 @@ def v1(run_with_both_engines_lora): pass +# The V1 lora test for this model requires more than 24GB. +@pytest.mark.skip_v1 @pytest.mark.xfail(current_platform.is_rocm(), reason="There can be output mismatch on ROCm") def test_gemma_lora(gemma_lora_files):