fix(github actions): pin torch version to 2.4

Modalities · Nov 12, 2024 · da5862e · da5862e
1 parent 787099d
commit da5862e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/tests_full.yml b/.github/workflows/tests_full.yml
@@ -23,7 +23,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install curl -y       # required by coveralls
         sudo apt-get install git -y
-        python -m pip install torch
+        python -m pip install torch~=2.4.1
         python -m pip install --upgrade pip setuptools wheel
         export FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE
         python -m pip install -e .[tests]

diff --git a/README.md b/README.md
@@ -59,7 +59,7 @@ Currently, the flash attention dependency cannot be installed without torch bein
 Until the flash attention developers fix this, we have to run
 
 ```sh
-pip install torch
+pip install torch~=2.4.1
 ```
 beforehand.
 
@@ -75,7 +75,7 @@ pip install -e .
 To install Modalities via pip, run
 
 ```sh
-pip install torch
+pip install torch~=2.4.1
 pip install modalities
 ```
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ description = "Modalities, a PyTorch-native framework for distributed and reprod
 readme = "README.md"
 dependencies = [
     "numpy<2.0",
-    "torch>=2.3",
+    "torch~=2.4.1",
     "packaging",
     "tqdm",
     "pyyaml",