"vscode:/vscode.git/clone" did not exist on "b99b1617cf5f0cc92547c2b11a41f27403c00147"
Unverified Commit cb8f3d90 authored by Johnny's avatar Johnny Committed by GitHub
Browse files

[NVIDIA] update pyproject.toml to support cu130 option (#11521)

parent 4b694e7d
......@@ -19,7 +19,7 @@ jobs:
version: 12.9.1
- runner: arm-docker-build-node
platform: linux/arm64
build_type: all_aarch64
build_type: all
tag: dev-arm64
version: 12.9.1
steps:
......
......@@ -66,7 +66,7 @@ jobs:
matrix:
variant:
- cuda_version: "12.9.1"
build_type: "all_aarch64"
build_type: "all"
runs-on: arm-docker-build-node
steps:
- name: Delete huge unnecessary tools folder
......
......@@ -46,7 +46,7 @@ dependencies = [
"py-spy",
"pybase64",
"pydantic",
"pynvml",
"nvidia-ml-py",
"python-multipart",
"pyzmq>=25.1.2",
"requests",
......@@ -73,7 +73,7 @@ dependencies = [
]
[project.optional-dependencies]
decord = ["decord"]
decord = ["decord2"]
test = [
"accelerate",
"expecttest",
......@@ -93,11 +93,21 @@ tracing = [
"opentelemetry-sdk",
]
all = ["sglang[test]", "sglang[decord]"]
all_aarch64 = ["sglang[test]"]
dev = ["sglang[test]", "sglang[decord]"]
cu130 = [
"torch==2.9.0",
"torchaudio==2.9.0",
"torchvision==0.24.0",
]
cu130_all = [
"sglang[test]",
"sglang[decord]",
"sglang[cu130]"
]
# The following will be deprecated in 2 weeks
dev = ["sglang[test]", "sglang[decord]"]
all_aarch64 = ["sglang[test]"]
blackwell = ["sglang[test]", "sglang[decord]"]
blackwell_aarch64 = ["sglang[test]"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment