IanNathaniel commited on
Commit
ed54751
1 Parent(s): 64548d1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  torch
 
2
  transformers
3
  cuda
 
1
  torch
2
+ torchvision
3
  transformers
4
  cuda