ping fa2

PrimeIntellect-ai · Dec 5, 2024 · 9863531 · 9863531
1 parent 963742e
commit 9863531
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -66,7 +66,7 @@ sudo apt install iperf -y
 uv venv
 source .venv/bin/activate
 uv sync --extra all
-uv pip install flash-attn --no-build-isolation
+uv pip install flash-attn==2.6.3 --no-build-isolation
 git submodule update --init --recursive
 ```
 

diff --git a/scripts/install/install.sh b/scripts/install/install.sh
@@ -48,7 +48,7 @@ main() {
     uv sync --extra all
 
     log_info "Installing flash-attn..."
-    uv pip install flash-attn --no-build-isolation
+    uv pip install flash-attn==2.6.3 --no-build-isolation
 
     log_info "Updating git submodules..."
     git submodule update --init --recursive