Fix chanves

1edb4521 · Logan Adams · 1dfc7ff0 · 1edb4521
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

tests/unit/inference/test_inference.py tests/unit/inference/test_inference.py +5 -0

未找到文件。
--- a/tests/unit/inference/test_inference.py
+++ b/tests/unit/inference/test_inference.py
@@ -389,6 +389,11 @@ class TestMPSize(DistributedTest):
                           "text-generation")],
                         ids=["gpt-j"])
 class TestLowCpuMemUsage(DistributedTest):
+    world_size = 1
+
+    def test(
+        self,
+        model_w_task,
        dtype,
        query,
        inf_kwargs,