Skip to content

Commit a48c9e2

Browse files
committed
Drop redundant ART megatron extra
1 parent 8d7a9f7 commit a48c9e2

2 files changed

Lines changed: 66 additions & 1539 deletions

File tree

pyproject.toml

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -41,20 +41,6 @@ backend = [
4141
"nvidia-cudnn-frontend<1.21 ; sys_platform == 'linux'",
4242
"vllm @ https://github.com/vivekkalyan/vllm/releases/download/v0.17.0-art1/vllm-0.17.0%2Bart1-cp38-abi3-manylinux_2_31_x86_64.whl ; sys_platform == 'linux'",
4343
]
44-
megatron = [
45-
"torch>=2.8.0",
46-
"quack-kernels==0.2.5",
47-
"apex",
48-
"transformer-engine==2.11.0",
49-
"transformer-engine-cu12==2.11.0",
50-
"transformer-engine-torch==2.11.0",
51-
"megatron-core==0.16.0rc0",
52-
"pybind11>=2.13.6",
53-
"megatron-bridge",
54-
"nvidia-ml-py==13.580.82",
55-
"ml-dtypes>=0.5.0 ; python_full_version < '3.13'",
56-
]
57-
5844
langgraph = [
5945
"langchain-core>=0.3.51",
6046
"langgraph>=0.6.2",

0 commit comments

Comments
 (0)