compile for torch 2.4.1 too

OpenProteinAI · Feb 7, 2025 · d2dd0fd · d2dd0fd
1 parent 4f8b153
commit d2dd0fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -44,7 +44,7 @@ jobs:
           # manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
           os: [ubuntu-20.04]
           python-version: ['3.9', '3.10']
-          torch-version: ['2.5.1']
+          torch-version: ['2.4.1', '2.5.1']
           cuda-version: ['11.8.0']
           # We need separate wheels that either uses C++11 ABI (-D_GLIBCXX_USE_CXX11_ABI) or not.
           # Pytorch wheels currently don't use it, but nvcr images have Pytorch compiled with C++11 ABI.