From 907782c237f8e1275d44311aecac45f285830456 Mon Sep 17 00:00:00 2001 From: rasbt Date: Wed, 22 Jan 2025 13:56:57 -0600 Subject: [PATCH] update --- .github/workflows/basic-tests-linux-pytorch-rc.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/basic-tests-linux-pytorch-rc.yml b/.github/workflows/basic-tests-linux-pytorch-rc.yml index 319dd9c6..376b9530 100644 --- 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: |