Skip to content

Commit

Permalink
disable calling torchrun in the zero to thunder notebook (Lightning-A…
Browse files Browse the repository at this point in the history
  • Loading branch information
t-vi authored May 27, 2024
1 parent 6f53464 commit 11e6143
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion notebooks/zero_to_thunder.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -3389,7 +3389,8 @@
}
],
"source": [
"!torchrun --nproc_per_node=2 zero_to_thunder_fsdp_simple_example.py"
"# commented out for CI limitations, see https://github.com/Lightning-AI/lightning-thunder/issues/465\n",
"# !torchrun --standalone --nnodes=1 --nproc_per_node=2 zero_to_thunder_fsdp_simple_example.py"
]
},
{
Expand Down

0 comments on commit 11e6143

Please sign in to comment.