Show tqdm progress bar
diff --git a/README.md b/README.md
index 6dd972314f..65b077f26c 100644
--- a/README.md
+++ b/README.md
@@ -264,7 +264,8 @@ torchrun --nproc_per_node=4 --master_port=20001 fastchat/train/train_mem.py \
     --tf32 True \
     --model_max_length 2048 \
     --gradient_checkpointing True \
-    --lazy_preprocess True
+    --lazy_preprocess True \
+    --disable_tqdm False
 ```
 
 - If you meet out-of-memory due to "FSDP Warning: When using FSDP, it is efficient and recommended... ", see solutions [here](https://github.com/huggingface/transformers/issues/24724#issuecomment-1645189539).
diff --git a/fastchat/train/llama_flash_attn_monkey_patch.py b/fastchat/train/llama_flash_attn_monkey_patch.py
index 031546502e..9b3392ed62 100644
--- a/fastchat/train/llama_flash_attn_monkey_patch.py
+++ b/fastchat/train/llama_flash_attn_monkey_patch.py
@@ -8,8 +8,9 @@
 from transformers.models.llama.modeling_llama import apply_rotary_pos_emb
 
 from einops import rearrange
-
-from flash_attn.flash_attn_interface import flash_attn_varlen_qkvpacked_func
+from flash_attn.flash_attn_interface import (  # pip3 install "flash-attn>=2.0"
+    flash_attn_varlen_qkvpacked_func,
+)
 from flash_attn.bert_padding import unpad_input, pad_input
 
 
diff --git a/scripts/test_readme_train.sh b/scripts/test_readme_train.sh
index 769f2c678a..6c2c7019e7 100644
--- a/scripts/test_readme_train.sh
+++ b/scripts/test_readme_train.sh
@@ -21,4 +21,5 @@ torchrun --nproc_per_node=4 --master_port=20001 fastchat/train/train_mem.py \
     --tf32 True \
     --model_max_length 2048 \
     --gradient_checkpointing True \
-    --lazy_preprocess True
+    --lazy_preprocess True \
+    --disable_tqdm False
