update

README.md

@@ -20,7 +20,7 @@ To use `fasckpt` with `flash_attn`, import and run `replace_hf_ckpt_with_fast_ck
 ```python
 # add monkey patch for fastckpt
 from fastckpt.llama_flash_attn_ckpt_monkey_patch import replace_hf_ckpt_with_fast_ckpt
-replace_llama_attn_with_flash_attn()
+replace_hf_ckpt_with_fast_ckpt()
 
 # import transformers and other packages
 import transformers