Update PyTorch versions to ones compatible with Numpy 2

pplantinga · pplantinga · commit c8e357a8d62f · 2025-03-23T16:50:21.000-04:00
diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml
@@ -30,7 +30,7 @@ jobs:
               run: |
                   pip install uv
                   uv pip install --system ctc-segmentation sacrebleu # ctc-segmentation is funky with uv due to their oldest-supported-numpy dependency
-                  uv pip install --system -r requirements.txt torch==2.2.1+cpu torchaudio==2.2.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu k2==1.24.4.dev20240223+cpu.torch2.2.1 --find-links https://k2-fsa.github.io/k2/cpu.html kaldilm==1.15.1 spacy==3.7.4 flair==0.13.1 gensim==4.3.2
+                  uv pip install --system -r requirements.txt torch==2.6.0+cpu torchaudio==2.6.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu k2==1.24.4.dev20250307+cpu.torch2.6.0 --find-links https://k2-fsa.github.io/k2/cpu.html kaldilm==1.15.1 spacy==3.7.4 flair==0.13.1 gensim==4.3.2
                   uv pip install --system --editable . --no-deps  # already installed pinned deps from requirements.txt, we're good
             - name: Install sox
               run: |
diff --git a/requirements.txt b/requirements.txt
@@ -10,7 +10,7 @@ pygtrie>=2.1,<3.0
 scipy>=1.4.1
 sentencepiece>=0.1.91
 SoundFile; sys_platform == 'win32'
-torch>=2.1.0
-torchaudio>=2.1.0
+torch>=2.3.0
+torchaudio>=2.3.0
 tqdm>=4.42.0
 transformers>=4.30.0
