diff --git a/.github/workflows/test-ui.yaml b/.github/workflows/test-ui.yaml
index 292ff5c6..3e96ac18 100644
--- a/.github/workflows/test-ui.yaml
+++ b/.github/workflows/test-ui.yaml
@@ -16,7 +16,7 @@ jobs:
     - name: Install requirements
       run: |
         python -m pip install --upgrade pip
-        pip install -r requirements.txt
+        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu -r requirements.txt
     - name: Run Tests
       run: | 
         npm ci