diff --git a/progen2/requirements.txt b/progen2/requirements.txt index 06142f3..ea0e739 100644 --- a/progen2/requirements.txt +++ b/progen2/requirements.txt @@ -1,4 +1,4 @@ --find-links https://download.pytorch.org/whl/torch_stable.html torch==1.9.0+cu111 -transformers==4.16.2 +transformers==4.30.0 tokenizers==0.10.3 \ No newline at end of file