HongFangzhou commited on
Commit
1002db9
Β·
1 Parent(s): 2117cc4

update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -2,6 +2,7 @@
2
  torch==1.12.0+cu113
3
  torchvision==0.13.0+cu113
4
  torchaudio==0.12.0
 
5
  numpy
6
  tqdm
7
  PyYAML
 
2
  torch==1.12.0+cu113
3
  torchvision==0.13.0+cu113
4
  torchaudio==0.12.0
5
+ pytorch-lightning==2.0.2
6
  numpy
7
  tqdm
8
  PyYAML