From 77071f201a9c10471dae9154096858bbd2a636a1 Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Tue, 6 Aug 2024 16:24:55 +0200 Subject: [PATCH] Fix torch version --- .github/workflows/ci.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index b57e820..bf442d1 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -25,6 +25,7 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip + pip install torch==2.2.0 # We require this specific version for numerical tests! python -m pip install --upgrade --no-cache-dir -e '.[dev]' - name: Log into HuggingFace run: |