We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 299e16a commit fa9bb3aCopy full SHA for fa9bb3a
1 file changed
tests/test_utils/runners/setup_training_test_env.sh
@@ -111,6 +111,7 @@ setup_cuda_training_env() {
111
datasets==4.5.0 \
112
fast-hadamard-transform==1.1.0 \
113
git+https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git@v0.2.0
114
+}
115
116
setup_metax_training_env() {
117
local megatron_dir="/tmp/Megatron-LM-FL"
0 commit comments