Update requirements.txt

requirements.txt

@@ -24,20 +24,18 @@ mdurl==0.1.2
 mpmath==1.3.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
 orjson==3.11.0
 packaging==25.0
 pandas==2.3.1
@@ -56,19 +54,18 @@ rich==14.1.0
 ruff==0.12.5
 safetensors==0.5.3
 semantic-version==2.10.0
-setuptools==80.9.0
 shellingham==1.5.4
 six==1.17.0
 sniffio==1.3.1
 starlette==0.47.2
-sympy==1.14.0
+sympy==1.13.1
 tokenizers==0.21.2
 tomlkit==0.12.0
-torch==2.7.1
-torchaudio==2.7.1
+torch==2.5.1
+torchaudio==2.5.1
 tqdm==4.67.1
 transformers==4.53.3
-triton==3.3.1
+triton==3.1.0
 typer==0.16.0
 typing-extensions==4.14.1
 tzdata==2025.2