Update dependencies

pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
     "smolagents[litellm,mcp,openai]>=1.17.0",
     "supervision>=0.25.1",
     "timm>=1.0.15",
-    "torch>=2.7.1",
+    "torch==2.5.1",
     "transformers>=4.52.4",
 ]
 

requirements.txt

@@ -78,20 +78,18 @@ multiprocess==0.70.16
 mypy-extensions==1.1.0
 networkx==3.4.2
 numpy==2.2.6
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-cupti-cu12==12.6.80
-nvidia-cuda-nvrtc-cu12==12.6.77
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.5.1.17
-nvidia-cufft-cu12==11.3.0.4
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.7.77
-nvidia-cusolver-cu12==11.7.1.2
-nvidia-cusparse-cu12==12.5.4.2
-nvidia-cusparselt-cu12==0.6.3
-nvidia-nccl-cu12==2.26.2
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.6.77
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
 openai==1.84.0
 opencv-python==4.11.0.86
 orjson==3.10.18
@@ -129,7 +127,6 @@ scikit-learn==1.7.0
 scipy==1.15.3
 semantic-version==2.10.0
 sentence-transformers==4.1.0
-setuptools==80.9.0
 shellingham==1.5.4
 sigtools==4.0.1
 six==1.17.0
@@ -139,7 +136,7 @@ sqlalchemy==2.0.41
 sse-starlette==2.3.6
 starlette==0.46.2
 supervision==0.25.1
-sympy==1.14.0
+sympy==1.13.1
 synchronicity==0.9.13
 tenacity==9.1.2
 threadpoolctl==3.6.0
@@ -148,11 +145,11 @@ timm==1.0.15
 tokenizers==0.21.1
 toml==0.10.2
 tomlkit==0.13.3
-torch==2.7.1
-torchvision==0.22.1
+torch==2.5.1
+torchvision==0.20.1
 tqdm==4.67.1
 transformers==4.52.4
-triton==3.3.1
+triton==3.1.0
 typer==0.16.0
 types-certifi==2021.10.8.3
 types-toml==0.10.8.20240310