MohamedRashad commited on
Commit
80ea65e
·
1 Parent(s): 661f80a

Update requirements.txt to specify torch version and add additional package sources

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,9 +1,12 @@
 
 
 
1
  accelerate
2
  sentencepiece
3
  diffusers
4
  gradio_client
5
  huggingface-hub==0.26.5
6
- torch
7
  torchvision==0.19.0
8
  pillow==10.4.0
9
  imageio==2.36.1
@@ -19,7 +22,6 @@ xatlas==0.0.9
19
  pyvista==0.44.2
20
  pymeshfix==0.17.0
21
  igraph==0.11.8
22
- git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
23
  xformers==0.0.27.post2
24
  kaolin==0.17.0
25
  spconv-cu120==2.3.6
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ --find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu121.html
3
+
4
  accelerate
5
  sentencepiece
6
  diffusers
7
  gradio_client
8
  huggingface-hub==0.26.5
9
+ torch==2.4.0
10
  torchvision==0.19.0
11
  pillow==10.4.0
12
  imageio==2.36.1
 
22
  pyvista==0.44.2
23
  pymeshfix==0.17.0
24
  igraph==0.11.8
 
25
  xformers==0.0.27.post2
26
  kaolin==0.17.0
27
  spconv-cu120==2.3.6