diff --git a/examples/minimal/ShardTensorExamples/6_ring_attention/README.md b/examples/minimal/ShardTensorExamples/6_ring_attention/README.md index 4877f4a144..37c8738368 100644 --- a/examples/minimal/ShardTensorExamples/6_ring_attention/README.md +++ b/examples/minimal/ShardTensorExamples/6_ring_attention/README.md @@ -37,6 +37,8 @@ torchrun --nproc-per-node 4 benchmark_sharded_attention.py \ After collecting JSON results in `results/`, generate scaling plots: +> Note! To run the scaling plot, you will need matplotlib installed. `pip install matplotlib`. + ```bash python plot_scaling_results.py ```