paulpanwang commited on
Commit
100f38d
·
verified ·
1 Parent(s): c8ad8c0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -1
requirements.txt CHANGED
@@ -1,9 +1,14 @@
 
 
 
 
 
 
1
  plyfile
2
  ipython
3
  numpy==1.26.4
4
  matplotlib
5
  Pillow
6
- imageio
7
  imageio-ffmpeg
8
  rembg[gpu]
9
  lpips
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+
3
+ torch==2.4.0
4
+ torchvision==0.19.0
5
+ pillow==10.4.0
6
+ imageio==2.36.1
7
  plyfile
8
  ipython
9
  numpy==1.26.4
10
  matplotlib
11
  Pillow
 
12
  imageio-ffmpeg
13
  rembg[gpu]
14
  lpips