File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -162,7 +162,7 @@ megatron = [
162162 " megatron-core==0.17.0; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
163163 " mbridge @ git+https://github.com/ISEEKYAN/mbridge.git@310e8fb; sys_platform == 'linux' and platform_machine == 'x86_64'" ,
164164 " megatron-bridge==0.4.0; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
165- " emerging-optimizers @ git+https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
165+ " emerging-optimizers @ git+https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git@v0.2.0 ; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
166166]
167167# Convenience extra for CUDA training packages (no inference backend)
168168cuda-train = [
Original file line number Diff line number Diff line change @@ -174,7 +174,7 @@ megatron = [
174174 " megatron-core==0.17.0; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
175175 " mbridge @ git+https://github.com/ISEEKYAN/mbridge.git@310e8fb; sys_platform == 'linux' and platform_machine == 'x86_64'" ,
176176 " megatron-bridge==0.4.0; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
177- " emerging-optimizers @ git+https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
177+ " emerging-optimizers @ git+https://github.com/NVIDIA-NeMo/Emerging-Optimizers.git@v0.2.0 ; python_version >= '3.12' and sys_platform == 'linux' and platform_machine == 'x86_64'" ,
178178]
179179cuda-train = [
180180 " areal[tms]" ,
You can’t perform that action at this time.
0 commit comments