JackWong0911 commited on
Commit
e0ae9a9
1 Parent(s): 04b5589

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  opencv-python
2
  imutils
3
  numpy
4
- torch
5
- torchvision
6
  pytorchvideo
7
  transformers
 
1
  opencv-python
2
  imutils
3
  numpy
4
+ torch==2.1.0
5
+ torchvision==0.16.0
6
  pytorchvideo
7
  transformers