Update requirements.txt

requirements.txt

@@ -1,5 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu121
-accelerate>=0.33.0
+--extra-index-url https://download.pytorch.org/whl/cu128
 aiofiles==23.2.1
 aiohappyeyeballs==2.6.1
 aiohttp==3.13.2
@@ -74,7 +73,7 @@ langid==1.1.6
 language-tags==1.2.0
 lazy_loader==0.4
 librosa
-llvmlite==0.42.0
+llvmlite
 loguru==0.7.3
 markdown-it-py==4.0.0
 MarkupSafe
@@ -86,29 +85,15 @@ multidict==6.7.0
 multiprocess==0.70.18
 networkx==3.1
 num2words==0.5.13
-numba==0.59.0
-numpy==1.26.0
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-cusparselt-cu12==0.6.3
-nvidia-nccl-cu12==2.20.5
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.1.105
+numba==0.60.0
+numpy==2.0.2
 omegaconf==2.3.0
 onnx==1.16.0
 onnxruntime
 onnxruntime-gpu
 orjson==3.11.5
 packaging==25.0
-pandas==2.3.3
+pandas==2.2.3
 phonemizer==3.3.0
 pillow==11.3.0
 platformdirs==4.5.1
@@ -162,13 +147,13 @@ threadpoolctl==3.6.0
 tokenizers==0.22.1
 tomli
 tomlkit==0.13.3
-torch==2.3.1
-torchaudio==2.3.1
-torchdiffeq==0.2.4
+torch==2.8.0
+torchaudio==2.8.0
+torchdiffeq
 tqdm>=4.65.0
 transformers
 transformers-stream-generator
-triton==2.3.1
+triton
 typer==0.16.0
 typing_extensions==4.12.2
 tzdata==2025.3