diff --git a/tests/diffusion/kernels/mot/test_mot_norm.py b/tests/diffusion/kernels/mot/test_mot_norm.py index c3b8bf53d..af6724d1e 100644 --- a/tests/diffusion/kernels/mot/test_mot_norm.py +++ b/tests/diffusion/kernels/mot/test_mot_norm.py @@ -22,6 +22,8 @@ from vllm.model_executor.layers.layernorm import rms_norm from vllm_omni.diffusion.layers.mot.mot_layernorm import MoTRMSNorm +pytestmark = [pytest.mark.core_model, pytest.mark.diffusion, pytest.mark.gpu] + # --------------------------------------------------------------------------- # Constants # ---------------------------------------------------------------------------