Hugging Face's logo

Spaces:
TencentARC
/
DI-PCG
Running on Zero

App Files Community
1
thuzhaowang commited on
Commit
7ad39d9
·
1 Parent(s): cbf8a2b

fix xformers

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -4
requirements.txt CHANGED
@@ -1,8 +1,11 @@
1
  rembg
2
  onnxruntime-gpu==1.17
3
- torch==2.4.0 --index-url https://download.pytorch.org/whl/cu118
4
- torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu118
5
- torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118
 
 
 
6
  accelerate==0.34.2
7
  bpy==3.6.0
8
  Cython==3.0.11
@@ -34,5 +37,6 @@ tqdm==4.66.5
34
  transformers==4.36.0
35
  trimesh==4.4.4
36
  triton==3.0.0
37
- https://download.pytorch.org/whl/cu118/xformers-0.0.27.post1%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl
 
38
  git+https://github.com/princeton-vl/[email protected]
 
1
  rembg
2
  onnxruntime-gpu==1.17
3
+ #torch==2.4.0 --index-url https://download.pytorch.org/whl/cu118
4
+ #torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu118
5
+ #torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118
6
+ torch==2.4.0 --index-url https://download.pytorch.org/whl/cu121
7
+ torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu121
8
+ torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121
9
  accelerate==0.34.2
10
  bpy==3.6.0
11
  Cython==3.0.11
 
37
  transformers==4.36.0
38
  trimesh==4.4.4
39
  triton==3.0.0
40
+ xformers===0.0.27.post2
41
+ #https://download.pytorch.org/whl/cu118/xformers-0.0.27.post1%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl
42
  git+https://github.com/princeton-vl/[email protected]