提交 6c1ec096 编写于 作者: L Logan Adams

Formatting has changes

上级 00f47f03
......@@ -426,7 +426,7 @@ class TestLowCpuMemUsage(DistributedTest):
print(local_rank, "deepspeed", ds_output)
assert assert_fn(bs_output, ds_output)
@pytest.mark.parametrize("model_w_task", [("tiiuae/falcon-7b", "text-generation")], ids=["falcon"])
class TestAutoTP(DistributedTest):
world_size = 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册