[NVIDIA] update pyproject.toml to support cu130 option (#11521)
This commit is contained in:
2
.github/workflows/release-docker-dev.yml
vendored
2
.github/workflows/release-docker-dev.yml
vendored
@@ -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:
|
||||
|
||||
2
.github/workflows/release-docker.yml
vendored
2
.github/workflows/release-docker.yml
vendored
@@ -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]"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user