Gagan Bhatia commited on
Commit
16a453e
1 Parent(s): 5db72a9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -2,6 +2,7 @@ numpy==1.21.1
2
  datasets==1.10.2
3
  pytorch_lightning==1.3.5
4
  transformers==4.9.0
 
5
  dagshub==0.1.6
6
  rouge_score
7
 
 
2
  datasets==1.10.2
3
  pytorch_lightning==1.3.5
4
  transformers==4.9.0
5
+ torch==1.9.0
6
  dagshub==0.1.6
7
  rouge_score
8