Skip to content

Commit d6c2a9e

Browse files
fix: require nvidia-resiliency-ext>=0.6.0 for updated megatron-core
megatron-core's has_nvrx_async_support asserts a minimum nvidia-resiliency-ext version of 0.6.0, which conflicts with the override cap of <0.6.0 and breaks import/collection in unit tests.
1 parent 83144b0 commit d6c2a9e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ override-dependencies = [
149149
"datasets>=3.3.0",
150150
"flash-linear-attention>=0.3.0,<0.4.dev0",
151151
"patchelf; sys_platform=='never'",
152-
"nvidia-resiliency-ext>=0.3.0,<0.6.0",
152+
"nvidia-resiliency-ext>=0.6.0",
153153
"transformer-engine[pytorch,core_cu13]>=2.14.0a0,<2.17.0; sys_platform != 'darwin'",
154154
"transformer-engine-cu13>=2.14.0a0,<2.17.0; sys_platform != 'darwin'",
155155
"transformer-engine-cu12; sys_platform == 'never'",

0 commit comments

Comments
 (0)