diff --git a/requirements.txt b/requirements.txt index 0a2397e..20db809 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ --find-links https://download.pytorch.org/whl/torch_stable.html accelerate>=0.25.0 -torch==2.1.2+cu118 +torch==2.2.0 torchvision==0.16.2+cu118 transformers>=4.30.0 numpy>=1.23.5