badayvedat commited on
Commit
8e68749
1 Parent(s): 5681680

Update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  torch==2.0.1
3
  torchaudio==2.0.2
 
4
  gdown
5
  lightning==2.0.0
6
  transformers==4.28.1
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  torch==2.0.1
3
  torchaudio==2.0.2
4
+ torchvision==0.15.2
5
  gdown
6
  lightning==2.0.0
7
  transformers==4.28.1