modelscope
diff --git a/‎pyproject.toml‎
Lines changed: 1 addition & 1 deletion b/‎pyproject.toml‎
Lines changed: 1 addition & 1 deletion
@@ -24,7 +24,7 @@ transformers = [
   "torchvision",
 ]
 kernels = ["kernels"]
-megatron = ["megatron-core>=0.12.0", "transformer-engine[pytorch]"]
+megatron = ["megatron-core>=0.12.0", "transformer-engine[pytorch]", "mcore_bridge"]
 vllm = ["vllm>=0.11"]
 ray = ["ray[serve]"]
 tinker = ["tinker==0.14.0"]
Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@ transformers = [`
`24`	`24`	`"torchvision",`
`25`	`25`	`]`
`26`	`26`	`kernels = ["kernels"]`
`27`		`-megatron = ["megatron-core>=0.12.0", "transformer-engine[pytorch]"]`
	`27`	`+megatron = ["megatron-core>=0.12.0", "transformer-engine[pytorch]", "mcore_bridge"]`
`28`	`28`	`vllm = ["vllm>=0.11"]`
`29`	`29`	`ray = ["ray[serve]"]`
`30`	`30`	`tinker = ["tinker==0.14.0"]`