Skip to content

Commit 26bec77

Browse files
committed
add missing Torchrun, which was missed during merge
Signed-off-by: ansjindal <[email protected]>
1 parent 4cdf6f6 commit 26bec77

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

nemo_run/core/execution/skypilot.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
Executor,
2828
ExecutorMacros,
2929
)
30+
from nemo_run.core.execution.launcher import FaultTolerance, Torchrun
3031
from nemo_run.core.packaging.base import Packager
3132
from nemo_run.core.packaging.git import GitArchivePackager
3233

0 commit comments

Comments
 (0)