update FA2 version

huggingface · Jan 24, 2024 · 906477b · 906477b
1 parent 7b38326
commit 906477b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/3d_parallelism_unit_tests.yaml b/.github/workflows/3d_parallelism_unit_tests.yaml
@@ -48,7 +48,6 @@ jobs:
         python -m pip install --upgrade pip
         pip install packaging
         pip install wheel
-        pip install "flash-attn>=2.4.2"  --no-build-isolation
         git clone https://github.com/huggingface/nanotron.git
         cd nanotron
         pip install -e .

diff --git a/.github/workflows/fa2_unit_tests.yaml b/.github/workflows/fa2_unit_tests.yaml
@@ -43,7 +43,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install packaging
         pip install wheel
-        pip install "flash-attn>=2.4.2"  --no-build-isolation
+        pip install flash-attn  --no-build-isolation
         git clone https://github.com/huggingface/nanotron.git
         cd nanotron
         pip install -e .