Skip to content

Commit e7eb579

Browse files
committed
fix using CPU torch with UV
1 parent 33d748d commit e7eb579

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

.github/workflows/ci-tests-fabric.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ jobs:
6666
TORCH_URL_TEST: "https://download.pytorch.org/whl/test/cpu/"
6767
# TODO: Remove this - Enable running MPS tests on this platform
6868
DISABLE_MPS: ${{ matrix.os == 'macOS-14' && '1' || '0' }}
69+
UV_TORCH_BACKEND: "cpu"
6970
steps:
7071
- uses: actions/checkout@v5
7172

.github/workflows/ci-tests-pytorch.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ jobs:
7171
FREEZE_REQUIREMENTS: ${{ ! (github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/heads/release/')) }}
7272
# TODO: Remove this - Enable running MPS tests on this platform
7373
DISABLE_MPS: ${{ matrix.os == 'macOS-14' && '1' || '0' }}
74+
UV_TORCH_BACKEND: "cpu"
7475
steps:
7576
- uses: actions/checkout@v5
7677

0 commit comments

Comments
 (0)