yisol commited on
Commit
373ec36
Β·
verified Β·
1 Parent(s): bd7f06a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -1,7 +1,6 @@
1
- transformers==4.36.2
2
- torch
3
- torchvision
4
- torchaudio
5
  numpy==1.24.4
6
  scipy==1.10.1
7
  scikit-image==0.21.0
@@ -20,4 +19,5 @@ av
20
  fvcore
21
  cloudpickle
22
  omegaconf
23
- pycocotools
 
 
1
+ torch==2.1.0
2
+ torchvision==0.16.0
3
+ torchaudio==2.1.0
 
4
  numpy==1.24.4
5
  scipy==1.10.1
6
  scikit-image==0.21.0
 
19
  fvcore
20
  cloudpickle
21
  omegaconf
22
+ pycocotools
23
+ huggingface_hub==0.25.0