diff --git a/requirements.txt b/requirements.txt index 9ac8f4f..74b59ee 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ pytorch-lightning==1.8.5 pyyaml>=6.0.1 tokenizers>=0.19,<0.20 --extra-index-url https://download.pytorch.org/whl/cu117 -torch==1.13.1+cu117 +torch==2.7.0 torchvision==0.14.1+cu117 tqdm>=4.65.0 transformers @ git+https://github.com/huggingface/transformers.git@v4.42.4