From 309aaa459f582f77272fb882a43601c6c5200c65 Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Tue, 22 Aug 2023 13:46:13 -0700 Subject: [PATCH] Update tests/unit/inference/test_inference.py --- tests/unit/inference/test_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/inference/test_inference.py b/tests/unit/inference/test_inference.py index da5ad81e..e591a214 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 -- GitLab