Unverified Commit 9aa33a74 authored by Charlie Fu's avatar Charlie Fu Committed by GitHub
Browse files

[Rocm][CI] Fix test_speculator_eagle3 by skipping the CompressedTensorw4a16 Model (#30001)


Signed-off-by: default avatarcharlifu <charlifu@amd.com>
Co-authored-by: default avatarAlexei-V-Ivanov-AMD <156011006+Alexei-V-Ivanov-AMD@users.noreply.github.com>
parent fd68e909
...@@ -5,6 +5,7 @@ import torch ...@@ -5,6 +5,7 @@ import torch
from vllm.config import SpeculativeConfig from vllm.config import SpeculativeConfig
from vllm.model_executor.models.interfaces import supports_eagle3 from vllm.model_executor.models.interfaces import supports_eagle3
from vllm.platforms import current_platform
@pytest.mark.parametrize( @pytest.mark.parametrize(
...@@ -21,6 +22,10 @@ from vllm.model_executor.models.interfaces import supports_eagle3 ...@@ -21,6 +22,10 @@ from vllm.model_executor.models.interfaces import supports_eagle3
pytest.param( pytest.param(
"nm-testing/Speculator-Qwen3-8B-Eagle3-converted-071-quantized-w4a16", "nm-testing/Speculator-Qwen3-8B-Eagle3-converted-071-quantized-w4a16",
id="qwen3-eagle3-speculator-w4a16-verifier", id="qwen3-eagle3-speculator-w4a16-verifier",
marks=pytest.mark.skipif(
current_platform.is_rocm(),
reason="The tests are skipped on rocm platform.",
),
), ),
], ],
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment