We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8d7a9f7 commit a48c9e2Copy full SHA for a48c9e2
2 files changed
pyproject.toml
@@ -41,20 +41,6 @@ backend = [
41
"nvidia-cudnn-frontend<1.21 ; sys_platform == 'linux'",
42
"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'",
43
]
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
-
58
langgraph = [
59
"langchain-core>=0.3.51",
60
"langgraph>=0.6.2",
0 commit comments