[shadformer] fix doc test cmd;

hpcaitech · Dec 25, 2024 · ee6bba9 · ee6bba9
1 parent 52a3b88
commit ee6bba9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/en/features/zerobubble_pipeline_parallelism.md b/docs/source/en/features/zerobubble_pipeline_parallelism.md
@@ -235,4 +235,4 @@ Performance Benchmark
 ## API Reference
 {{ autodoc:colossalai.pipeline.schedule.zero_bubble_pp.ZeroBubbleVPipeScheduler }}
 
-<!-- doc-test-command: python -m pytest -s tests/test_pipeline/test_schedule/test_zerobubble_pp.py  -->
+<!-- doc-test-command: torchrun --standalone --nproc_per_node=4 zerobubble_pipeline_parallelism.py  -->
diff --git a/docs/source/zh-Hans/features/zerobubble_pipeline_parallelism.md b/docs/source/zh-Hans/features/zerobubble_pipeline_parallelism.md
@@ -234,4 +234,4 @@ plugin = HybridParallelPlugin(
 ## API 参考
 {{ autodoc:colossalai.pipeline.schedule.zero_bubble_pp.ZeroBubbleVPipeScheduler }}
 
-<!-- doc-test-command: python -m pytest -s tests/test_pipeline/test_schedule/test_zerobubble_pp.py  -->
+<!-- doc-test-command: torchrun --standalone --nproc_per_node=4 zerobubble_pipeline_parallelism.py  -->