From fc8325c07b8bc52294d56811e5baddeb8ed99283 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Tue, 20 Aug 2024 15:30:45 -0700
Subject: [PATCH] install only cpu versions

---
 .github/workflows/python-test.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml
index 0da3b3b..3a42ee9 100644
--- a/.github/workflows/python-test.yml
+++ b/.github/workflows/python-test.yml
@@ -28,6 +28,7 @@ jobs:
         python -m pip install --upgrade pip
         python -m pip install pytest
         python -m pip install wheel
+        python -m pip install torch==2.4.0+cpu torchvision==0.19.0+cpu
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Test with pytest
       run: |