From 8f71dc8cd9ac549f55131461f6d29f1b39ccefa5 Mon Sep 17 00:00:00 2001 From: Casper Date: Sat, 16 Nov 2024 20:36:11 +0100 Subject: [PATCH] Minimum of torch 2.2.0 during build (#649) --- .github/workflows/build.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 53c8fba7..35ded640 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -44,6 +44,8 @@ jobs: defaults: run: shell: pwsh + env: + MIN_TORCH_VER: "2.2.0" steps: - name: Free Disk Space @@ -78,7 +80,7 @@ jobs: - name: Install Dependencies run: | # Install torch - python -m pip install --upgrade --no-cache-dir torch==2.5.1 + python -m pip install --upgrade --no-cache-dir torch==$env:MIN_TORCH_VER python -m pip install build setuptools wheel # Print version information