Cognomen commited on
Commit
4392731
Β·
1 Parent(s): 30b098a

change torch and torchvision ver

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -6,5 +6,5 @@ opencv-python
6
  transformers
7
  flax
8
  pillow
9
- torch
10
- torchvision
 
6
  transformers
7
  flax
8
  pillow
9
+ torch==1.12.1+cu113
10
+ torchvision==0.13.1+cu113