suolyer commited on
Commit
c004869
1 Parent(s): 2ecf9b9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
- numpy==1.23.4
3
  torch
 
4
  pytorch-lightning==1.6.0
5
  transformers==4.20.0
6
  streamlit==1.10.0
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
 
2
  torch
3
+ torchvision
4
  pytorch-lightning==1.6.0
5
  transformers==4.20.0
6
  streamlit==1.10.0