ChuxiJ commited on
Commit
a7270bf
·
1 Parent(s): 6922ca4

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 @ {root:uri}/acestep/third_parts/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
+ ]