diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index d6d174f20..ac434716a 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -73,7 +73,7 @@ jobs:
       - name: Install dependencies
         run: |
           uv venv
-          uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+          uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           uv pip install -e ".[dev]"
 
       - name: Install transformers