Skip to content

Commit 23daa3b

Browse files
committed
chore: add emerging-optimizers version
1 parent 5c545e6 commit 23daa3b

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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)
168168
cuda-train = [

pyproject.vllm.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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
]
179179
cuda-train = [
180180
"areal[tms]",

0 commit comments

Comments
 (0)