fix torch

requirements.txt

@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu126
+--extra-index-url https://download.pytorch.org/whl/cu124
 certifi==2025.1.31
 charset-normalizer==3.4.1
 cupy-cuda12x==13.3.0
@@ -38,9 +38,9 @@ safetensors==0.5.2
 scipy==1.15.1
 sympy==1.13.1
 timm==1.0.14
-torch==2.4.0+cu126
-torchaudio==2.6.0+cu126
-torchvision==0.21.0+cu126
+torch==2.4.0
+torchvision==0.19.0
+torchaudio==2.4.0
 tqdm==4.67.1
 triton==3.2.0
 typing_extensions==4.12.2