[fix] fix test_shard_llama

2026-05-16 20:25:37 +00:00 · 2024-10-25 09:01:13 +00:00
parent d0ec221b38
commit cc0dfddcbc
3 changed files with 24 additions and 19 deletions
--- a/examples/language/llama/benchmark.py
+++ b/examples/language/llama/benchmark.py
@@ -237,7 +237,6 @@ def main():
            ).get_v_schedule()
        else:
            scheduler_nodes = None
-        # print(f"{dist.get_rank()} {scheduler_nodes[]} ")

        plugin = HybridParallelPlugin(
            tp_size=args.tp,