slobers commited on
Commit
99bf892
·
verified ·
1 Parent(s): 9adb6e5

Update pyproject.toml

Browse files
Files changed (1) hide show
  1. pyproject.toml +8 -4
pyproject.toml CHANGED
@@ -1,5 +1,5 @@
1
  [build-system]
2
- requires = ["setuptools >= 75.0"]
3
  build-backend = "setuptools.build_meta"
4
 
5
  [project]
@@ -16,13 +16,17 @@ dependencies = [
16
  "xformers==0.0.25.post1",
17
  "triton==2.2.0",
18
  "transformers==4.41.2",
 
19
  "omegaconf==2.3.0",
20
  "torch==2.2.2",
21
  "torchvision==0.17.2",
22
- "edge-maxxing-pipelines @ git+https://github.com/womboai/edge-maxxing@e7c760ac54f6052803dadb3ade8ebfc9679a94589#subdirectory=pipelines",
23
  "huggingface-hub==0.25.2",
24
  "oneflow",
25
  "setuptools>=75.2.0",
 
 
 
26
  ]
27
 
28
  [tool.edge-maxxing]
@@ -33,7 +37,7 @@ models = [
33
 
34
  [tool.uv.sources]
35
  oneflow = { url = "https://github.com/siliconflow/oneflow_releases/releases/download/community_cu118/oneflow-0.9.1.dev20240802%2Bcu118-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
 
36
 
37
  [project.scripts]
38
- start_inference = "main:main"
39
-
 
1
  [build-system]
2
+ requires = ["setuptools >= 61.0"]
3
  build-backend = "setuptools.build_meta"
4
 
5
  [project]
 
16
  "xformers==0.0.25.post1",
17
  "triton==2.2.0",
18
  "transformers==4.41.2",
19
+ "accelerate==0.31.0",
20
  "omegaconf==2.3.0",
21
  "torch==2.2.2",
22
  "torchvision==0.17.2",
23
+ "edge-maxxing-pipelines @ git+https://github.com/womboai/edge-maxxing@8d8ff45863416484b5b4bc547782591bbdfc696a#subdirectory=pipelines",
24
  "huggingface-hub==0.25.2",
25
  "oneflow",
26
  "setuptools>=75.2.0",
27
+ "bitsandbytes>=0.44.1",
28
+ "stable-fast",
29
+ "tomesd>=0.1.3",
30
  ]
31
 
32
  [tool.edge-maxxing]
 
37
 
38
  [tool.uv.sources]
39
  oneflow = { url = "https://github.com/siliconflow/oneflow_releases/releases/download/community_cu118/oneflow-0.9.1.dev20240802%2Bcu118-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
40
+ stable-fast = { url = "https://github.com/chengzeyi/stable-fast/releases/download/v1.0.5/stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl" }
41
 
42
  [project.scripts]
43
+ start_inference = "main:main"