diff --git a/requirements.txt b/requirements.txt index a98ef2a..f9d3a78 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ transformers @ git+http://github.com/ibeltagy/transformers.git@longformer_encoder_decoder#egg=transformers -pytorch-lightning @ git+http://github.com/ibeltagy/pytorch-lightning.git@v0.8.5_fixes#egg=pytorch-lightning +pytorch-lightning==1.0.4 torch==1.6.0 tensorboardX test-tube==0.7.5