diff --git a/requirements.txt b/requirements.txt index 512e832..52088ef 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ pandas sentencepiece torch>=1.7.0,!=1.8.0 -transformers==4.10.0 -pytorch-lightning==1.4.5 +transformers>=4.10.0,<4.11.0 +pytorch-lightning>=1.4.5,<1.5.0