Sort pyproject dependency lists (#28811)
This commit is contained in:
+26
-25
@@ -1,5 +1,5 @@
|
|||||||
[build-system]
|
[build-system]
|
||||||
requires = ["setuptools>=61.0", "setuptools-scm>=8.0", "setuptools-rust>=1.10", "wheel"]
|
requires = ["setuptools>=61.0", "setuptools-rust>=1.10", "setuptools-scm>=8.0", "wheel"]
|
||||||
build-backend = "setuptools.build_meta"
|
build-backend = "setuptools.build_meta"
|
||||||
|
|
||||||
[project]
|
[project]
|
||||||
@@ -14,35 +14,40 @@ classifiers = [
|
|||||||
"License :: OSI Approved :: Apache Software License",
|
"License :: OSI Approved :: Apache Software License",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# Please keep dependency lists in this file sorted alphabetically by package name.
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"IPython",
|
|
||||||
"aiohttp",
|
"aiohttp",
|
||||||
"apache-tvm-ffi==0.1.9",
|
|
||||||
"anthropic>=0.20.0",
|
"anthropic>=0.20.0",
|
||||||
|
"apache-tvm-ffi==0.1.9",
|
||||||
|
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
||||||
"blobfile==3.0.0",
|
"blobfile==3.0.0",
|
||||||
"build",
|
"build",
|
||||||
"compressed-tensors",
|
"compressed-tensors",
|
||||||
"cuda-python>=13.0",
|
"cuda-python>=13.0",
|
||||||
"decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
|
||||||
"datasets",
|
"datasets",
|
||||||
|
"decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
||||||
"distro",
|
"distro",
|
||||||
|
"easydict", # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports
|
||||||
"einops",
|
"einops",
|
||||||
"fastapi",
|
"fastapi",
|
||||||
"flashinfer_python[cu13]==0.6.12", # keep it aligned with jit-cache version in Dockerfile
|
"flash-attn-4>=4.0.0b9",
|
||||||
"flashinfer_cubin==0.6.12",
|
"flashinfer_cubin==0.6.12",
|
||||||
|
"flashinfer_python[cu13]==0.6.12", # keep it aligned with jit-cache version in Dockerfile
|
||||||
"gguf",
|
"gguf",
|
||||||
"interegular",
|
"interegular",
|
||||||
|
"IPython",
|
||||||
|
"kernels>=0.14.1,<0.15",
|
||||||
"llguidance>=0.7.11,<0.8.0",
|
"llguidance>=0.7.11,<0.8.0",
|
||||||
|
"mistral_common>=1.11.0",
|
||||||
"modelscope",
|
"modelscope",
|
||||||
"msgspec",
|
"msgspec",
|
||||||
"ninja",
|
"ninja",
|
||||||
"easydict", # Required by remote model code (e.g. DeepSeek-OCR) loaded via trust_remote_code; validated by transformers 5.4+ check_imports
|
|
||||||
"numpy",
|
"numpy",
|
||||||
"nvidia-cutlass-dsl[cu13]==4.5.2",
|
"nvidia-cutlass-dsl[cu13]==4.5.2",
|
||||||
"nvidia-ml-py",
|
|
||||||
"nvidia-mathdx==25.6.0",
|
"nvidia-mathdx==25.6.0",
|
||||||
"openai-harmony==0.0.4",
|
"nvidia-ml-py",
|
||||||
"openai==2.6.1",
|
"openai==2.6.1",
|
||||||
|
"openai-harmony==0.0.4",
|
||||||
"orjson",
|
"orjson",
|
||||||
"outlines==0.1.11",
|
"outlines==0.1.11",
|
||||||
"packaging",
|
"packaging",
|
||||||
@@ -60,30 +65,26 @@ dependencies = [
|
|||||||
"scipy",
|
"scipy",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"setproctitle",
|
"setproctitle",
|
||||||
"flash-attn-4>=4.0.0b9",
|
|
||||||
"sgl-deep-gemm==0.1.3",
|
"sgl-deep-gemm==0.1.3",
|
||||||
"sglang-kernel==0.4.4",
|
"sglang-kernel==0.4.4",
|
||||||
|
"smg-grpc-servicer>=0.5.0",
|
||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
"tilelang==0.1.8",
|
"tilelang==0.1.8",
|
||||||
"timm==1.0.16",
|
"timm==1.0.16",
|
||||||
"tokenspeed_mla==0.1.6",
|
"tokenspeed_mla==0.1.6",
|
||||||
"torch_memory_saver>=0.0.9.post1",
|
|
||||||
"torch==2.11.0",
|
"torch==2.11.0",
|
||||||
|
"torch_memory_saver>=0.0.9.post1",
|
||||||
"torchao==0.17.0",
|
"torchao==0.17.0",
|
||||||
"torchaudio==2.11.0",
|
"torchaudio==2.11.0",
|
||||||
"torchcodec==0.11.1 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec 0.11.1 for torch 2.11.x (0.10 is ABI-incompatible: references the pre-2.11 c10::MessageLogger ctor signature). Not available on Linux ARM.
|
"torchcodec==0.11.1 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec 0.11.1 for torch 2.11.x (0.10 is ABI-incompatible: references the pre-2.11 c10::MessageLogger ctor signature). Not available on Linux ARM.
|
||||||
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
|
||||||
"torchvision",
|
"torchvision",
|
||||||
"tqdm",
|
"tqdm",
|
||||||
"mistral_common>=1.11.0",
|
|
||||||
"transformers==5.8.1",
|
"transformers==5.8.1",
|
||||||
"uvicorn",
|
"uvicorn",
|
||||||
"uvloop",
|
"uvloop",
|
||||||
"watchfiles",
|
"watchfiles",
|
||||||
"xgrammar==0.2.1",
|
"xgrammar==0.2.1",
|
||||||
"smg-grpc-servicer>=0.5.0",
|
|
||||||
"kernels>=0.14.1,<0.15",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[[tool.uv.index]]
|
[[tool.uv.index]]
|
||||||
@@ -95,7 +96,9 @@ default = true
|
|||||||
checkpoint-engine = ["checkpoint-engine==0.1.2"]
|
checkpoint-engine = ["checkpoint-engine==0.1.2"]
|
||||||
runai = ["runai-model-streamer[s3,gcs,azure]>=0.15.7"]
|
runai = ["runai-model-streamer[s3,gcs,azure]>=0.15.7"]
|
||||||
diffusion = [
|
diffusion = [
|
||||||
"PyYAML==6.0.1",
|
"addict==2.4.0",
|
||||||
|
"av==16.1.0",
|
||||||
|
"cache-dit==1.3.0",
|
||||||
"cloudpickle==3.1.2",
|
"cloudpickle==3.1.2",
|
||||||
"diffusers==0.37.0",
|
"diffusers==0.37.0",
|
||||||
"imageio==2.36.0",
|
"imageio==2.36.0",
|
||||||
@@ -103,15 +106,13 @@ diffusion = [
|
|||||||
"moviepy>=2.0.0",
|
"moviepy>=2.0.0",
|
||||||
"nvidia-modelopt",
|
"nvidia-modelopt",
|
||||||
"opencv-python-headless==4.10.0.84",
|
"opencv-python-headless==4.10.0.84",
|
||||||
|
"PyYAML==6.0.1",
|
||||||
"remote-pdb==2.1.0",
|
"remote-pdb==2.1.0",
|
||||||
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
|
||||||
"vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
|
||||||
"runai_model_streamer>=0.15.7",
|
"runai_model_streamer>=0.15.7",
|
||||||
"cache-dit==1.3.0",
|
|
||||||
"addict==2.4.0",
|
|
||||||
"av==16.1.0",
|
|
||||||
"scikit-image==0.25.2",
|
"scikit-image==0.25.2",
|
||||||
|
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||||
"trimesh>=4.0.0",
|
"trimesh>=4.0.0",
|
||||||
|
"vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||||
"websockets",
|
"websockets",
|
||||||
"xatlas",
|
"xatlas",
|
||||||
]
|
]
|
||||||
@@ -139,7 +140,9 @@ test = [
|
|||||||
"accelerate",
|
"accelerate",
|
||||||
"addict",
|
"addict",
|
||||||
"bitsandbytes",
|
"bitsandbytes",
|
||||||
|
"diff-cover",
|
||||||
"expecttest",
|
"expecttest",
|
||||||
|
"granian>=2.6.0",
|
||||||
"jsonlines",
|
"jsonlines",
|
||||||
"lm-eval[api]>=0.4.9.2",
|
"lm-eval[api]>=0.4.9.2",
|
||||||
"matplotlib",
|
"matplotlib",
|
||||||
@@ -149,24 +152,22 @@ test = [
|
|||||||
"polars",
|
"polars",
|
||||||
"pytest",
|
"pytest",
|
||||||
"pytest-cov",
|
"pytest-cov",
|
||||||
"diff-cover",
|
|
||||||
"sentence_transformers",
|
"sentence_transformers",
|
||||||
"sglang[fastokens]",
|
"sglang[fastokens]",
|
||||||
"tabulate",
|
"tabulate",
|
||||||
"granian>=2.6.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
dev = ["sglang[test]"]
|
dev = ["sglang[test]"]
|
||||||
|
|
||||||
all = [
|
all = [
|
||||||
"sglang[diffusion]",
|
"sglang[diffusion]",
|
||||||
"sglang[tracing]",
|
|
||||||
"sglang[http2]",
|
"sglang[http2]",
|
||||||
|
"sglang[tracing]",
|
||||||
]
|
]
|
||||||
|
|
||||||
[tool.uv.extra-build-dependencies]
|
[tool.uv.extra-build-dependencies]
|
||||||
st-attn = ["torch", "setuptools"]
|
st-attn = ["setuptools", "torch"]
|
||||||
vsa = ["torch", "setuptools"]
|
vsa = ["setuptools", "torch"]
|
||||||
|
|
||||||
[project.urls]
|
[project.urls]
|
||||||
"Homepage" = "https://github.com/sgl-project/sglang"
|
"Homepage" = "https://github.com/sgl-project/sglang"
|
||||||
|
|||||||
+13
-12
@@ -15,27 +15,29 @@ classifiers = [
|
|||||||
"License :: OSI Approved :: Apache Software License",
|
"License :: OSI Approved :: Apache Software License",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# Please keep dependency lists in this file sorted alphabetically by package name.
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"IPython",
|
|
||||||
"aiohttp",
|
"aiohttp",
|
||||||
"anthropic>=0.20.0",
|
"anthropic>=0.20.0",
|
||||||
"blobfile==3.0.0",
|
"blobfile==3.0.0",
|
||||||
"build",
|
"build",
|
||||||
"compressed-tensors",
|
"compressed-tensors",
|
||||||
"datasets",
|
"datasets",
|
||||||
|
"easydict",
|
||||||
"einops",
|
"einops",
|
||||||
"fastapi",
|
"fastapi",
|
||||||
"gguf",
|
"gguf",
|
||||||
"intel-openmp; platform_machine == 'x86_64'",
|
"intel-openmp; platform_machine == 'x86_64'",
|
||||||
"interegular",
|
"interegular",
|
||||||
|
"IPython",
|
||||||
"llguidance>=0.7.11,<0.8.0",
|
"llguidance>=0.7.11,<0.8.0",
|
||||||
|
"mistral_common>=1.11.0",
|
||||||
"modelscope",
|
"modelscope",
|
||||||
"msgspec",
|
"msgspec",
|
||||||
"easydict",
|
|
||||||
"ninja",
|
"ninja",
|
||||||
"numpy",
|
"numpy",
|
||||||
"openai-harmony==0.0.4",
|
|
||||||
"openai==2.6.1",
|
"openai==2.6.1",
|
||||||
|
"openai-harmony==0.0.4",
|
||||||
"orjson",
|
"orjson",
|
||||||
"outlines",
|
"outlines",
|
||||||
"packaging",
|
"packaging",
|
||||||
@@ -52,6 +54,7 @@ dependencies = [
|
|||||||
"scipy",
|
"scipy",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"setproctitle",
|
"setproctitle",
|
||||||
|
"smg-grpc-servicer>=0.5.0",
|
||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tabulate",
|
"tabulate",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
@@ -61,41 +64,39 @@ dependencies = [
|
|||||||
"torchaudio==2.11.0",
|
"torchaudio==2.11.0",
|
||||||
"torchvision==0.27.0",
|
"torchvision==0.27.0",
|
||||||
"tqdm",
|
"tqdm",
|
||||||
"mistral_common>=1.11.0",
|
|
||||||
"transformers==5.8.1",
|
"transformers==5.8.1",
|
||||||
"triton==3.7.0",
|
"triton==3.7.0",
|
||||||
"uvicorn",
|
"uvicorn",
|
||||||
"uvloop",
|
"uvloop",
|
||||||
"xgrammar==0.2.1",
|
"xgrammar==0.2.1",
|
||||||
"smg-grpc-servicer>=0.5.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.optional-dependencies]
|
[project.optional-dependencies]
|
||||||
diffusion = [
|
diffusion = [
|
||||||
"PyYAML==6.0.1",
|
"addict==2.4.0",
|
||||||
|
"av==16.1.0",
|
||||||
|
"cache-dit==1.3.0",
|
||||||
"cloudpickle==3.1.2",
|
"cloudpickle==3.1.2",
|
||||||
"diffusers==0.37.0",
|
"diffusers==0.37.0",
|
||||||
"imageio==2.36.0",
|
"imageio==2.36.0",
|
||||||
"imageio-ffmpeg==0.5.1",
|
"imageio-ffmpeg==0.5.1",
|
||||||
"moviepy>=2.0.0",
|
"moviepy>=2.0.0",
|
||||||
"opencv-python-headless==4.10.0.84",
|
"opencv-python-headless==4.10.0.84",
|
||||||
|
"PyYAML==6.0.1",
|
||||||
"remote-pdb==2.1.0",
|
"remote-pdb==2.1.0",
|
||||||
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
|
||||||
"vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
|
||||||
"runai_model_streamer>=0.15.5",
|
"runai_model_streamer>=0.15.5",
|
||||||
"cache-dit==1.3.0",
|
|
||||||
"addict==2.4.0",
|
|
||||||
"av==16.1.0",
|
|
||||||
"scikit-image==0.25.2",
|
"scikit-image==0.25.2",
|
||||||
|
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||||
"trimesh>=4.0.0",
|
"trimesh>=4.0.0",
|
||||||
|
"vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||||
"xatlas",
|
"xatlas",
|
||||||
]
|
]
|
||||||
|
|
||||||
tracing = [
|
tracing = [
|
||||||
"opentelemetry-sdk",
|
|
||||||
"opentelemetry-api",
|
"opentelemetry-api",
|
||||||
"opentelemetry-exporter-otlp",
|
"opentelemetry-exporter-otlp",
|
||||||
"opentelemetry-exporter-otlp-proto-grpc",
|
"opentelemetry-exporter-otlp-proto-grpc",
|
||||||
|
"opentelemetry-sdk",
|
||||||
]
|
]
|
||||||
test = [
|
test = [
|
||||||
"accelerate",
|
"accelerate",
|
||||||
|
|||||||
+20
-19
@@ -14,30 +14,32 @@ classifiers = [
|
|||||||
"License :: OSI Approved :: Apache Software License",
|
"License :: OSI Approved :: Apache Software License",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# Please keep dependency lists in this file sorted alphabetically by package name.
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"IPython",
|
|
||||||
"aiohttp",
|
"aiohttp",
|
||||||
"anthropic>=0.20.0",
|
"anthropic>=0.20.0",
|
||||||
"blobfile==3.0.0",
|
|
||||||
"av",
|
"av",
|
||||||
|
"blobfile==3.0.0",
|
||||||
"build",
|
"build",
|
||||||
"compressed-tensors",
|
"compressed-tensors",
|
||||||
"datasets",
|
"datasets",
|
||||||
"decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
"decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
||||||
|
"easydict",
|
||||||
"einops",
|
"einops",
|
||||||
"fastapi",
|
"fastapi",
|
||||||
"easydict",
|
|
||||||
"gguf",
|
"gguf",
|
||||||
"hf_transfer",
|
"hf_transfer",
|
||||||
"huggingface_hub",
|
"huggingface_hub",
|
||||||
"interegular",
|
"interegular",
|
||||||
|
"IPython",
|
||||||
"llguidance>=0.7.11,<0.8.0",
|
"llguidance>=0.7.11,<0.8.0",
|
||||||
|
"mistral_common>=1.11.0",
|
||||||
"modelscope",
|
"modelscope",
|
||||||
"msgspec",
|
"msgspec",
|
||||||
"ninja",
|
"ninja",
|
||||||
"numpy",
|
"numpy",
|
||||||
"openai-harmony==0.0.4",
|
|
||||||
"openai==2.6.1",
|
"openai==2.6.1",
|
||||||
|
"openai-harmony==0.0.4",
|
||||||
"orjson",
|
"orjson",
|
||||||
"outlines==0.1.11",
|
"outlines==0.1.11",
|
||||||
"packaging",
|
"packaging",
|
||||||
@@ -54,35 +56,34 @@ dependencies = [
|
|||||||
"scipy",
|
"scipy",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"setproctitle",
|
"setproctitle",
|
||||||
|
"smg-grpc-servicer>=0.5.0",
|
||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
"timm==1.0.16",
|
"timm==1.0.16",
|
||||||
"torchao==0.9.0",
|
"torchao==0.9.0",
|
||||||
"tqdm",
|
"tqdm",
|
||||||
"mistral_common>=1.11.0",
|
|
||||||
"transformers==5.8.1",
|
"transformers==5.8.1",
|
||||||
"uvicorn",
|
"uvicorn",
|
||||||
"uvloop",
|
"uvloop",
|
||||||
"xgrammar==0.2.1",
|
"xgrammar==0.2.1",
|
||||||
"smg-grpc-servicer>=0.5.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.optional-dependencies]
|
[project.optional-dependencies]
|
||||||
checkpoint-engine = ["checkpoint-engine==0.1.2"]
|
checkpoint-engine = ["checkpoint-engine==0.1.2"]
|
||||||
diffusion = [
|
diffusion = [
|
||||||
"PyYAML==6.0.1",
|
"addict",
|
||||||
"cloudpickle",
|
"cache-dit==1.3.5",
|
||||||
"diffusers==0.37.0",
|
"cloudpickle",
|
||||||
"imageio==2.36.0",
|
"diffusers==0.37.0",
|
||||||
"imageio-ffmpeg==0.5.1",
|
"imageio==2.36.0",
|
||||||
"moviepy>=2.0.0",
|
"imageio-ffmpeg==0.5.1",
|
||||||
"opencv-python==4.10.0.84",
|
"moviepy>=2.0.0",
|
||||||
"remote-pdb",
|
"opencv-python==4.10.0.84",
|
||||||
"cache-dit==1.3.5",
|
"PyYAML==6.0.1",
|
||||||
"addict",
|
"remote-pdb",
|
||||||
"scikit-image==0.25.2",
|
"scikit-image==0.25.2",
|
||||||
"trimesh>=4.0.0",
|
"trimesh>=4.0.0",
|
||||||
"xatlas",
|
"xatlas",
|
||||||
]
|
]
|
||||||
|
|
||||||
tracing = [
|
tracing = [
|
||||||
|
|||||||
+34
-33
@@ -13,16 +13,16 @@ classifiers = [
|
|||||||
"Programming Language :: Python :: 3",
|
"Programming Language :: Python :: 3",
|
||||||
"License :: OSI Approved :: Apache Software License",
|
"License :: OSI Approved :: Apache Software License",
|
||||||
]
|
]
|
||||||
dependencies = ["aiohttp", "requests", "tqdm", "numpy", "IPython", "setproctitle"]
|
# Please keep dependency lists in this file sorted alphabetically by package name.
|
||||||
|
dependencies = ["aiohttp", "IPython", "numpy", "requests", "setproctitle", "tqdm"]
|
||||||
|
|
||||||
[project.optional-dependencies]
|
[project.optional-dependencies]
|
||||||
runtime_common = [
|
runtime_common = [
|
||||||
"IPython",
|
|
||||||
"aiohttp",
|
"aiohttp",
|
||||||
"apache-tvm-ffi",
|
|
||||||
"anthropic>=0.20.0",
|
"anthropic>=0.20.0",
|
||||||
"blobfile==3.0.0",
|
"apache-tvm-ffi",
|
||||||
"av",
|
"av",
|
||||||
|
"blobfile==3.0.0",
|
||||||
"build",
|
"build",
|
||||||
"compressed-tensors",
|
"compressed-tensors",
|
||||||
"datasets",
|
"datasets",
|
||||||
@@ -31,13 +31,15 @@ runtime_common = [
|
|||||||
"fastapi",
|
"fastapi",
|
||||||
"gguf",
|
"gguf",
|
||||||
"interegular",
|
"interegular",
|
||||||
|
"IPython",
|
||||||
"llguidance>=0.7.11,<0.8.0",
|
"llguidance>=0.7.11,<0.8.0",
|
||||||
|
"mistral_common>=1.11.0",
|
||||||
"modelscope",
|
"modelscope",
|
||||||
"msgspec",
|
"msgspec",
|
||||||
"ninja",
|
"ninja",
|
||||||
"numpy",
|
"numpy",
|
||||||
"openai-harmony==0.0.4",
|
|
||||||
"openai==2.6.1",
|
"openai==2.6.1",
|
||||||
|
"openai-harmony==0.0.4",
|
||||||
"orjson",
|
"orjson",
|
||||||
"outlines==0.1.11",
|
"outlines==0.1.11",
|
||||||
"packaging",
|
"packaging",
|
||||||
@@ -54,61 +56,60 @@ runtime_common = [
|
|||||||
"scipy",
|
"scipy",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"setproctitle",
|
"setproctitle",
|
||||||
|
"smg-grpc-servicer>=0.5.0",
|
||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
"timm==1.0.16",
|
"timm==1.0.16",
|
||||||
"torchao==0.9.0",
|
"torchao==0.9.0",
|
||||||
"tqdm",
|
"tqdm",
|
||||||
"mistral_common>=1.11.0",
|
|
||||||
"transformers==5.8.1",
|
"transformers==5.8.1",
|
||||||
"uvicorn",
|
"uvicorn",
|
||||||
"uvloop",
|
"uvloop",
|
||||||
"xgrammar==0.2.1",
|
"xgrammar==0.2.1",
|
||||||
"smg-grpc-servicer>=0.5.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
diffusion_common = [
|
diffusion_common = [
|
||||||
"PyYAML==6.0.1",
|
"addict",
|
||||||
"cloudpickle",
|
"cloudpickle",
|
||||||
"diffusers==0.37.0",
|
"diffusers==0.37.0",
|
||||||
"imageio==2.36.0",
|
"imageio==2.36.0",
|
||||||
"imageio-ffmpeg==0.5.1",
|
"imageio-ffmpeg==0.5.1",
|
||||||
"moviepy>=2.0.0",
|
"moviepy>=2.0.0",
|
||||||
"opencv-python-headless==4.10.0.84",
|
"opencv-python-headless==4.10.0.84",
|
||||||
|
"PyYAML==6.0.1",
|
||||||
"remote-pdb",
|
"remote-pdb",
|
||||||
"addict",
|
|
||||||
"scikit-image==0.25.2",
|
"scikit-image==0.25.2",
|
||||||
"trimesh>=4.0.0",
|
"trimesh>=4.0.0",
|
||||||
"xatlas",
|
"xatlas",
|
||||||
]
|
]
|
||||||
|
|
||||||
tracing = [
|
tracing = [
|
||||||
"opentelemetry-sdk",
|
|
||||||
"opentelemetry-api",
|
"opentelemetry-api",
|
||||||
"opentelemetry-exporter-otlp",
|
"opentelemetry-exporter-otlp",
|
||||||
"opentelemetry-exporter-otlp-proto-grpc",
|
"opentelemetry-exporter-otlp-proto-grpc",
|
||||||
|
"opentelemetry-sdk",
|
||||||
]
|
]
|
||||||
|
|
||||||
# HIP (Heterogeneous-computing Interface for Portability) for AMD
|
# HIP (Heterogeneous-computing Interface for Portability) for AMD
|
||||||
# => base docker rocm/vllm-dev:20250114, not from public vllm whl
|
# => base docker rocm/vllm-dev:20250114, not from public vllm whl
|
||||||
srt_hip = [
|
srt_hip = [
|
||||||
"sglang[runtime_common]",
|
|
||||||
"torch",
|
|
||||||
"petit_kernel==0.0.2",
|
|
||||||
"wave-lang==3.8.2",
|
|
||||||
# Pin to 0.15.0: 0.16.0 needs torch>=2.10 (incompatible with ROCm torch
|
# Pin to 0.15.0: 0.16.0 needs torch>=2.10 (incompatible with ROCm torch
|
||||||
# 2.9.1). An open-ended `<0.16.0` made pip backtrack into an unbuildable
|
# 2.9.1). An open-ended `<0.16.0` made pip backtrack into an unbuildable
|
||||||
# ancient setuptools sdist; an exact pin keeps the resolver converging.
|
# ancient setuptools sdist; an exact pin keeps the resolver converging.
|
||||||
"compressed-tensors==0.15.0",
|
"compressed-tensors==0.15.0",
|
||||||
|
"petit_kernel==0.0.2",
|
||||||
|
"sglang[runtime_common]",
|
||||||
|
"torch",
|
||||||
|
"wave-lang==3.8.2",
|
||||||
]
|
]
|
||||||
|
|
||||||
diffusion_hip = [
|
diffusion_hip = [
|
||||||
"sglang[diffusion_common]",
|
"cache-dit==1.3.0",
|
||||||
"peft>=0.18.0,<0.19.0", # Pin to <0.19.0 due to torchao incompatibility
|
"peft>=0.18.0,<0.19.0", # Pin to <0.19.0 due to torchao incompatibility
|
||||||
|
"runai_model_streamer>=0.15.5",
|
||||||
|
"sglang[diffusion_common]",
|
||||||
"st_attn==0.0.7",
|
"st_attn==0.0.7",
|
||||||
"vsa==0.0.4",
|
"vsa==0.0.4",
|
||||||
"runai_model_streamer>=0.15.5",
|
|
||||||
"cache-dit==1.3.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
# For Intel Gaudi(device : hpu) follow the installation guide
|
# For Intel Gaudi(device : hpu) follow the installation guide
|
||||||
@@ -117,44 +118,44 @@ srt_hpu = ["sglang[runtime_common]"]
|
|||||||
|
|
||||||
# https://docs.sglang.io/platforms/mthreads_gpu.md
|
# https://docs.sglang.io/platforms/mthreads_gpu.md
|
||||||
srt_musa = [
|
srt_musa = [
|
||||||
|
"deep-gemm>=0.1.3",
|
||||||
|
"flash_attn_3>=0.1.4",
|
||||||
|
"mate>=0.2.0",
|
||||||
|
"mthreads-ml-py",
|
||||||
|
"numpy<2.0",
|
||||||
"sglang[runtime_common]",
|
"sglang[runtime_common]",
|
||||||
"torch",
|
"torch",
|
||||||
"torch_musa",
|
"torch_musa",
|
||||||
"torchada>=0.1.59",
|
"torchada>=0.1.59",
|
||||||
"mthreads-ml-py",
|
|
||||||
"mate>=0.2.0",
|
|
||||||
"deep-gemm>=0.1.3",
|
|
||||||
"flash_attn_3>=0.1.4",
|
|
||||||
"numpy<2.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
diffusion_musa = [
|
diffusion_musa = [
|
||||||
|
"cache-dit==1.1.8",
|
||||||
|
"runai_model_streamer>=0.15.5",
|
||||||
"sglang[diffusion_common]",
|
"sglang[diffusion_common]",
|
||||||
"st_attn==0.0.7",
|
"st_attn==0.0.7",
|
||||||
"vsa==0.0.4",
|
"vsa==0.0.4",
|
||||||
"runai_model_streamer>=0.15.5",
|
|
||||||
"cache-dit==1.1.8",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
# https://docs.sglang.io/platforms/mps.md
|
# https://docs.sglang.io/platforms/mps.md
|
||||||
srt_mps = [
|
srt_mps = [
|
||||||
|
"mlx",
|
||||||
|
"mlx-lm",
|
||||||
"sglang[runtime_common]",
|
"sglang[runtime_common]",
|
||||||
"torch==2.11.0",
|
"torch==2.11.0",
|
||||||
"torchao==0.9.0",
|
"torchao==0.9.0",
|
||||||
"torchaudio==2.11.0",
|
"torchaudio==2.11.0",
|
||||||
"torchvision",
|
"torchvision",
|
||||||
"mlx",
|
|
||||||
"mlx-lm",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
diffusion_mps = [
|
diffusion_mps = [
|
||||||
"sglang[diffusion_common]",
|
|
||||||
"cloudpickle==3.1.2",
|
|
||||||
"remote-pdb==2.1.0",
|
|
||||||
"cache-dit==1.2.3",
|
|
||||||
"addict==2.4.0",
|
"addict==2.4.0",
|
||||||
"av==16.1.0",
|
"av==16.1.0",
|
||||||
|
"cache-dit==1.2.3",
|
||||||
|
"cloudpickle==3.1.2",
|
||||||
|
"remote-pdb==2.1.0",
|
||||||
"scikit-image==0.25.2",
|
"scikit-image==0.25.2",
|
||||||
|
"sglang[diffusion_common]",
|
||||||
"trimesh>=4.0.0",
|
"trimesh>=4.0.0",
|
||||||
"xatlas",
|
"xatlas",
|
||||||
]
|
]
|
||||||
@@ -172,10 +173,10 @@ test = [
|
|||||||
"tabulate",
|
"tabulate",
|
||||||
]
|
]
|
||||||
|
|
||||||
all_hip = ["sglang[srt_hip]", "sglang[diffusion_hip]", "sglang[tracing]"]
|
all_hip = ["sglang[diffusion_hip]", "sglang[srt_hip]", "sglang[tracing]"]
|
||||||
all_hpu = ["sglang[srt_hpu]"]
|
all_hpu = ["sglang[srt_hpu]"]
|
||||||
all_musa = ["sglang[srt_musa]", "sglang[diffusion_musa]"]
|
all_musa = ["sglang[diffusion_musa]", "sglang[srt_musa]"]
|
||||||
all_mps = ["sglang[srt_mps]", "sglang[diffusion_mps]"]
|
all_mps = ["sglang[diffusion_mps]", "sglang[srt_mps]"]
|
||||||
|
|
||||||
dev_hip = ["sglang[all_hip]", "sglang[test]"]
|
dev_hip = ["sglang[all_hip]", "sglang[test]"]
|
||||||
dev_hpu = ["sglang[all_hpu]", "sglang[test]"]
|
dev_hpu = ["sglang[all_hpu]", "sglang[test]"]
|
||||||
|
|||||||
+17
-16
@@ -14,33 +14,30 @@ classifiers = [
|
|||||||
"License :: OSI Approved :: Apache Software License",
|
"License :: OSI Approved :: Apache Software License",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# Please keep dependency lists in this file sorted alphabetically by package name.
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"torch==2.12.0+xpu",
|
"addict",
|
||||||
"torchcodec==0.12.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torch==2.12.0 on XPU uses torchcodec 0.12.0
|
|
||||||
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
|
||||||
"torchaudio==2.11.0+xpu",
|
|
||||||
"torchvision==0.27.0+xpu",
|
|
||||||
"sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git",
|
|
||||||
"IPython",
|
|
||||||
"aiohttp",
|
"aiohttp",
|
||||||
"anthropic>=0.20.0",
|
"anthropic>=0.20.0",
|
||||||
|
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
||||||
"blobfile==3.0.0",
|
"blobfile==3.0.0",
|
||||||
"build",
|
"build",
|
||||||
"compressed-tensors",
|
"compressed-tensors",
|
||||||
"addict",
|
|
||||||
"datasets",
|
"datasets",
|
||||||
"easydict",
|
"easydict",
|
||||||
"einops",
|
"einops",
|
||||||
"fastapi",
|
"fastapi",
|
||||||
"gguf",
|
"gguf",
|
||||||
"interegular",
|
"interegular",
|
||||||
|
"IPython",
|
||||||
"llguidance>=0.7.11,<0.8.0",
|
"llguidance>=0.7.11,<0.8.0",
|
||||||
|
"mistral_common>=1.11.0",
|
||||||
"modelscope",
|
"modelscope",
|
||||||
"msgspec",
|
"msgspec",
|
||||||
"ninja",
|
"ninja",
|
||||||
"numpy",
|
"numpy",
|
||||||
"openai-harmony==0.0.4",
|
|
||||||
"openai==2.6.1",
|
"openai==2.6.1",
|
||||||
|
"openai-harmony==0.0.4",
|
||||||
"orjson",
|
"orjson",
|
||||||
"outlines==0.1.11",
|
"outlines==0.1.11",
|
||||||
"packaging",
|
"packaging",
|
||||||
@@ -57,35 +54,39 @@ dependencies = [
|
|||||||
"scipy",
|
"scipy",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"setproctitle",
|
"setproctitle",
|
||||||
|
"sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git",
|
||||||
|
"smg-grpc-servicer>=0.5.0",
|
||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
"timm==1.0.16",
|
"timm==1.0.16",
|
||||||
|
"torch==2.12.0+xpu",
|
||||||
"torchao==0.17.0+xpu",
|
"torchao==0.17.0+xpu",
|
||||||
|
"torchaudio==2.11.0+xpu",
|
||||||
|
"torchcodec==0.12.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torch==2.12.0 on XPU uses torchcodec 0.12.0
|
||||||
|
"torchvision==0.27.0+xpu",
|
||||||
"tqdm",
|
"tqdm",
|
||||||
"mistral_common>=1.11.0",
|
|
||||||
"transformers==5.8.1",
|
"transformers==5.8.1",
|
||||||
"uvicorn",
|
"uvicorn",
|
||||||
"uvloop",
|
"uvloop",
|
||||||
# "xgrammar==0.2.1", xgrammar depends on CUDA PyTorch and Triton only
|
# "xgrammar==0.2.1", xgrammar depends on CUDA PyTorch and Triton only
|
||||||
"smg-grpc-servicer>=0.5.0",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.optional-dependencies]
|
[project.optional-dependencies]
|
||||||
diffusion = [
|
diffusion = [
|
||||||
"PyYAML==6.0.1",
|
"addict==2.4.0",
|
||||||
|
"av==16.1.0",
|
||||||
|
"cache-dit==1.3.0",
|
||||||
"cloudpickle==3.1.2",
|
"cloudpickle==3.1.2",
|
||||||
"diffusers==0.38.0",
|
"diffusers==0.38.0",
|
||||||
"imageio==2.36.0",
|
"imageio==2.36.0",
|
||||||
"imageio-ffmpeg==0.5.1",
|
"imageio-ffmpeg==0.5.1",
|
||||||
"moviepy>=2.0.0",
|
"moviepy>=2.0.0",
|
||||||
"opencv-python==4.10.0.84",
|
"opencv-python==4.10.0.84",
|
||||||
|
"PyYAML==6.0.1",
|
||||||
"remote-pdb==2.1.0",
|
"remote-pdb==2.1.0",
|
||||||
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
|
||||||
"runai_model_streamer>=0.15.5",
|
"runai_model_streamer>=0.15.5",
|
||||||
"cache-dit==1.3.0",
|
|
||||||
"addict==2.4.0",
|
|
||||||
"av==16.1.0",
|
|
||||||
"scikit-image==0.25.2",
|
"scikit-image==0.25.2",
|
||||||
|
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
|
||||||
"trimesh>=4.0.0",
|
"trimesh>=4.0.0",
|
||||||
"xatlas",
|
"xatlas",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -34,6 +34,7 @@ def run_server(server_args):
|
|||||||
|
|
||||||
asyncio.run(serve_grpc(server_args))
|
asyncio.run(serve_grpc(server_args))
|
||||||
elif server_args.use_ray:
|
elif server_args.use_ray:
|
||||||
|
# Ray mode: HTTP mode with Ray backend.
|
||||||
try:
|
try:
|
||||||
from sglang.srt.ray.http_server import launch_server
|
from sglang.srt.ray.http_server import launch_server
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
|||||||
Reference in New Issue
Block a user