Update requirements.txt

requirements.txt

@@ -1,7 +1,7 @@
 --find-links https://download.pytorch.org/whl/torch_stable.html
 accelerate>=0.25.0
-torch==2.1.2+cu118
-torchvision==0.16.2+cu118
+torch==2.0.1+cu118
+torchvision==0.15.2+cu118
 transformers>=4.30.0
 numpy>=1.23.5
 kornia>=0.7.0
@@ -16,4 +16,4 @@ onnxruntime>=1.16.3
 einops>=0.7.0
 onnx2torch>=1.5.13
 warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
-torchtools @ git+https://github.com/pabloppp/pytorch-tools
+torchtools @ git+https://github.com/pabloppp/pytorch-tools