update

rasbt · Jan 22, 2025 · 907782c · 907782c
1 parent df10ba3
commit 907782c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/basic-tests-linux-pytorch-rc.yml b/.github/workflows/basic-tests-linux-pytorch-rc.yml
@@ -1,4 +1,4 @@
-name: Code tests (Linux)
+name: Code tests (PyTorch RC)
 
 on:
   push:
@@ -36,6 +36,7 @@ jobs:
         pip install pytest nbval
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
         pip install -r ch05/07_gpt_to_llama/tests/test-requirements-extra.txt
+        pip install torch==2.6.0 torchvision --index-url https://download.pytorch.org/whl/test/cpu
 
     - name: Test Selected Python Scripts
       run: |