ParthRangarajan commited on
Commit
ca10564
1 Parent(s): dff0197

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
- torch
 
2
  transformers
3
- aitextgen
 
1
+ torch==1.9.0
2
+ pytorch-lightning==1.7.7
3
  transformers
4
+ aitextgen==0.6.0