fix: Update torch version to 2.6.0 to pass testcases

Signed-off-by: Akash-Nayak <akash19nayak@gmail.com>

Author: Akash-Nayak

pyproject.toml

@@ -28,9 +28,9 @@ classifiers=[
 ]
 dependencies = [
 "numpy>=1.26.4,<2.0",
-"accelerate==1.6.0",
+"accelerate>=0.20.3,!=0.34,<1.7",
 "transformers>=4.52.3",
-"torch==2.5.0",
+"torch==2.6.0",
 "sentencepiece>=0.1.99,<0.3",
 "tokenizers>=0.13.3,<1.0",
 "tqdm>=4.66.2,<5.0",