chore: Update PyTorch and torchvision versions in requirements.txt

- Upgraded torch from 2.1.2 to 2.5.1 and torchvision from 0.16.2 to 0.20.1 to ensure compatibility with new features and improvements.

requirements.txt

@@ -1,5 +1,5 @@
-torch==2.1.2
-torchvision==0.16.2
+torch==2.5.1
+torchvision==0.20.1
 accelerate==0.31.0
 git+https://github.com/huggingface/diffusers.git
 matplotlib==3.9.1