From 9863531beec394e77db9d88b9214ab147d325ad6 Mon Sep 17 00:00:00 2001
From: Sami Jaghouar <sami.jaghouar@hotmail.fr>
Date: Thu, 5 Dec 2024 07:10:00 +0000
Subject: [PATCH] ping fa2

---
 README.md                  | 2 +-
 scripts/install/install.sh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 6e09ec8..2bee50b 100644
--- 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
index 5a4361d..5c1480c 100755
--- 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