ginipick commited on
Commit
a125850
·
verified ·
1 Parent(s): 51f4c93

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -4
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
- pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
 
 
 
2
  onnxruntime-gpu==1.17.0
3
- torch==2.2.0
4
- torch_scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
5
  accelerate==0.25.0
6
  datasets
7
  diffusers>=0.27.0
@@ -23,7 +24,7 @@ transformers>=4.37.0
23
  trimesh
24
  typeguard
25
  wandb
26
- xformers==0.0.24
27
  ninja
28
  spaces
29
  gradio_client
 
1
+ torch==2.6.0
2
+ torchvision
3
+ torchaudio
4
+ pytorch3d
5
  onnxruntime-gpu==1.17.0
 
 
6
  accelerate==0.25.0
7
  datasets
8
  diffusers>=0.27.0
 
24
  trimesh
25
  typeguard
26
  wandb
27
+ xformers
28
  ninja
29
  spaces
30
  gradio_client