Spaces:
Running
on
Zero
Running
on
Zero
feat: update dependencies
Browse files
acestep/third_parts/nano-vllm/pyproject.toml
CHANGED
|
@@ -15,6 +15,8 @@ dependencies = [
|
|
| 15 |
"triton-windows>=3.0.0; sys_platform == 'win32'",
|
| 16 |
"triton>=3.0.0; sys_platform != 'win32'",
|
| 17 |
"transformers>=4.51.0",
|
|
|
|
|
|
|
| 18 |
"xxhash",
|
| 19 |
]
|
| 20 |
|
|
|
|
| 15 |
"triton-windows>=3.0.0; sys_platform == 'win32'",
|
| 16 |
"triton>=3.0.0; sys_platform != 'win32'",
|
| 17 |
"transformers>=4.51.0",
|
| 18 |
+
"flash-attn @ https://github.com/sdbds/flash-attention-for-windows/releases/download/2.8.3/flash_attn-2.8.3+cu128torch2.8.0cxx11abiFALSEfullbackward-cp311-cp311-win_amd64.whl ; sys_platform == 'win32' and python_version == '3.11' and platform_machine == 'AMD64'",
|
| 19 |
+
"flash-attn; sys_platform != 'win32'",
|
| 20 |
"xxhash",
|
| 21 |
]
|
| 22 |
|
pyproject.toml
CHANGED
|
@@ -10,12 +10,10 @@ dependencies = [
|
|
| 10 |
"torch>=2.9.1; sys_platform != 'darwin'",
|
| 11 |
"torchvision; sys_platform != 'darwin'",
|
| 12 |
"torchaudio>=2.9.1; sys_platform != 'darwin'",
|
| 13 |
-
|
| 14 |
# PyTorch for macOS (CPU / MPS)
|
| 15 |
"torch>=2.9.1; sys_platform == 'darwin'",
|
| 16 |
"torchvision; sys_platform == 'darwin'",
|
| 17 |
"torchaudio>=2.9.1; sys_platform == 'darwin'",
|
| 18 |
-
|
| 19 |
# Common dependencies
|
| 20 |
"transformers",
|
| 21 |
"diffusers",
|
|
@@ -28,11 +26,16 @@ dependencies = [
|
|
| 28 |
"accelerate>=1.12.0",
|
| 29 |
"fastapi>=0.110.0",
|
| 30 |
"uvicorn[standard]>=0.27.0",
|
| 31 |
-
|
|
|
|
|
|
|
| 32 |
# Local third-party packages
|
| 33 |
-
"nano-vllm
|
| 34 |
]
|
| 35 |
|
|
|
|
|
|
|
|
|
|
| 36 |
[project.scripts]
|
| 37 |
acestep = "acestep.acestep_v15_pipeline:main"
|
| 38 |
acestep-api = "acestep.api_server:main"
|
|
@@ -49,4 +52,11 @@ name = "pytorch"
|
|
| 49 |
url = "https://download.pytorch.org/whl/cu128"
|
| 50 |
|
| 51 |
[tool.hatch.build.targets.wheel]
|
| 52 |
-
packages = ["acestep"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
"torch>=2.9.1; sys_platform != 'darwin'",
|
| 11 |
"torchvision; sys_platform != 'darwin'",
|
| 12 |
"torchaudio>=2.9.1; sys_platform != 'darwin'",
|
|
|
|
| 13 |
# PyTorch for macOS (CPU / MPS)
|
| 14 |
"torch>=2.9.1; sys_platform == 'darwin'",
|
| 15 |
"torchvision; sys_platform == 'darwin'",
|
| 16 |
"torchaudio>=2.9.1; sys_platform == 'darwin'",
|
|
|
|
| 17 |
# Common dependencies
|
| 18 |
"transformers",
|
| 19 |
"diffusers",
|
|
|
|
| 26 |
"accelerate>=1.12.0",
|
| 27 |
"fastapi>=0.110.0",
|
| 28 |
"uvicorn[standard]>=0.27.0",
|
| 29 |
+
"numba>=0.63.1",
|
| 30 |
+
"vector-quantize-pytorch>=1.27.15",
|
| 31 |
+
"torchcodec>=0.9.1",
|
| 32 |
# Local third-party packages
|
| 33 |
+
"nano-vllm",
|
| 34 |
]
|
| 35 |
|
| 36 |
+
[tool.uv.sources]
|
| 37 |
+
nano-vllm = { path = "acestep/third_parts/nano-vllm" }
|
| 38 |
+
|
| 39 |
[project.scripts]
|
| 40 |
acestep = "acestep.acestep_v15_pipeline:main"
|
| 41 |
acestep-api = "acestep.api_server:main"
|
|
|
|
| 52 |
url = "https://download.pytorch.org/whl/cu128"
|
| 53 |
|
| 54 |
[tool.hatch.build.targets.wheel]
|
| 55 |
+
packages = ["acestep"]
|
| 56 |
+
|
| 57 |
+
[tool.uv.extra-build-dependencies]
|
| 58 |
+
flash-attn = [
|
| 59 |
+
"packaging",
|
| 60 |
+
"wheel",
|
| 61 |
+
{ requirement = "torch", match-runtime = true },
|
| 62 |
+
]
|