diff --git a/.github/workflows/basic-tests-pytorch-rc.yml b/.github/workflows/basic-tests-pytorch-rc.yml index f99c733a..bc211f1c 100644 --- a/.github/workflows/basic-tests-pytorch-rc.yml +++ b/.github/workflows/basic-tests-pytorch-rc.yml @@ -1,4 +1,4 @@ -name: Test latest PyTorch release candidate +name: Test latest PyTorch nightly / release candidate on: push: branches: [ main ] @@ -35,7 +35,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 + pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu - name: Test Selected Python Scripts run: |