diff --git a/tests/unit/inference/test_inference.py b/tests/unit/inference/test_inference.py index da5ad81e3e358bc032b3ef554745574ed814f3ee..e591a214c3f7a8af71f1714f8080aa55b91da203 100644 --- a/tests/unit/inference/test_inference.py +++ b/tests/unit/inference/test_inference.py @@ -383,7 +383,7 @@ class TestMPSize(DistributedTest): assert assert_fn(bs_output, ds_output) -@pytest.mark.seq_inference +@pytest.mark.inference @pytest.mark.parametrize("model_w_task", [("gpt2", "text-generation")], ids=["gpt2"]) class TestLowCpuMemUsage(DistributedTest): world_size = 1