poojagoudkarnakanti commited on
Commit
7e5fa83
·
verified ·
1 Parent(s): fb69165

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch>=2.0.0
2
- torchvision
3
  transformer-lens>=1.0.0
4
  transformers>=4.30.0
5
  sentence-transformers>=2.2.0
 
1
+ torch>=2.1.0
2
+ torchvision==0.16.0
3
  transformer-lens>=1.0.0
4
  transformers>=4.30.0
5
  sentence-transformers>=2.2.0