xinjie.wang commited on
Commit
ea0649a
·
1 Parent(s): 4add0d9
Files changed (1) hide show
  1. requirements.txt +17 -8
requirements.txt CHANGED
@@ -1,9 +1,10 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
 
2
 
3
 
4
- torch==2.1.0
5
- torchaudio==2.1.0
6
- torchvision==0.16.0
7
  xformers==0.0.22.post7
8
  dataclasses_json
9
  easydict
@@ -19,7 +20,7 @@ pymeshfix==0.17.0
19
  igraph==0.11.8
20
  pyvista==0.36.1
21
  openai==1.58.1
22
- spconv-cu118==2.3.8
23
  transformers==4.42.4
24
  gradio_litmodel3d==0.0.1
25
  # gradio==5.12.0
@@ -28,7 +29,7 @@ diffusers==0.31.0
28
  xatlas==0.0.9
29
  onnxruntime==1.20.1
30
  tenacity==8.2.2
31
- pytorch-lightning==2.1.0
32
  accelerate==0.33.0
33
  basicsr==1.4.2
34
  realesrgan==0.3.0
@@ -39,8 +40,16 @@ utils3d@git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c
39
  clip@git+https://github.com/openai/CLIP.git
40
  kolors@git+https://github.com/Kwai-Kolors/Kolors.git#egg=038818d
41
  segment-anything@git+https://github.com/facebookresearch/segment-anything.git#egg=dca509f
42
- https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
43
  https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
44
- https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
45
  https://github.com/nerfstudio-project/gsplat/releases/download/v1.5.0/gsplat-1.5.0+pt21cu118-cp310-cp310-linux_x86_64.whl
46
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
 
 
 
 
 
 
 
 
1
+ # --extra-index-url https://download.pytorch.org/whl/cu118
2
+ --extra-index-url https://download.pytorch.org/whl/cu121
3
 
4
 
5
+ # torch==2.1.0
6
+ # torchaudio==2.1.0
7
+ # torchvision==0.16.0
8
  xformers==0.0.22.post7
9
  dataclasses_json
10
  easydict
 
20
  igraph==0.11.8
21
  pyvista==0.36.1
22
  openai==1.58.1
23
+ # spconv-cu118==2.3.8
24
  transformers==4.42.4
25
  gradio_litmodel3d==0.0.1
26
  # gradio==5.12.0
 
29
  xatlas==0.0.9
30
  onnxruntime==1.20.1
31
  tenacity==8.2.2
32
+ # pytorch-lightning==2.1.0
33
  accelerate==0.33.0
34
  basicsr==1.4.2
35
  realesrgan==0.3.0
 
40
  clip@git+https://github.com/openai/CLIP.git
41
  kolors@git+https://github.com/Kwai-Kolors/Kolors.git#egg=038818d
42
  segment-anything@git+https://github.com/facebookresearch/segment-anything.git#egg=dca509f
43
+ # https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
44
  https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
45
+ # https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
46
  https://github.com/nerfstudio-project/gsplat/releases/download/v1.5.0/gsplat-1.5.0+pt21cu118-cp310-cp310-linux_x86_64.whl
47
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
48
+
49
+
50
+ torch==2.4.0
51
+ torchvision==0.19.0
52
+ pytorch-lightning==2.4.0
53
+ spconv-cu120==2.3.6
54
+ https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
55
+ https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true