jw2yang commited on
Commit
bf1344c
·
1 Parent(s): 087a536
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  torch==2.4.0
2
- torchvision==0.18.1
3
  pytorch-lightning>=1.0.8
4
  transformers @ git+https://github.com/jwyang/transformers.git@dev/jwyang-v4.44.1
5
  tokenizers>=0.15.0
 
1
  torch==2.4.0
2
+ torchvision
3
  pytorch-lightning>=1.0.8
4
  transformers @ git+https://github.com/jwyang/transformers.git@dev/jwyang-v4.44.1
5
  tokenizers>=0.15.0