diff --git a/README.md b/README.md
index 9676603..7778809 100644
--- a/README.md
+++ b/README.md
@@ -54,7 +54,7 @@ LLaVA-Mini is a unified large multimodal model that can support the understandin
   conda activate llavamini
   pip install -e .
   pip install -e ".[train]"
-  pip install flash-attn --no-build-isolation
+  pip install flash-attn==2.3.4 --no-build-isolation
   ```
 
 ### Command Interaction