SkalskiP commited on
Commit
a110ab6
1 Parent(s): 1817830
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,5 +1,8 @@
 
 
 
 
1
  numpy
2
  pillow
3
- torch
4
  gradio
5
  transformers
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch
3
+ torchvision
4
+
5
  numpy
6
  pillow
 
7
  gradio
8
  transformers