diff --git a/transformer-arch/requirements.txt b/transformer-arch/requirements.txt index 6f564fa..883c44f 100644 --- a/transformer-arch/requirements.txt +++ b/transformer-arch/requirements.txt @@ -93,7 +93,7 @@ termcolor==1.1.0 threadpoolctl==3.0.0 tokenizers==0.10.3 --find-links https://download.pytorch.org/whl/torch_stable.html -torch==1.9.1+cu111 +torch==1.13.1 torchaudio==0.9.1 torchvision==0.10.1+cu111 tqdm==4.62.3