diff --git a/kit/extract_lightning_ckpt.py b/kit/extract_lightning_ckpt.py
index 5c7381c..736d375 100644
--- a/kit/extract_lightning_ckpt.py
+++ b/kit/extract_lightning_ckpt.py
@@ -5,8 +5,8 @@
 from exp.run_pretrain import Wrapper
 
 if __name__ == "__main__":
-    ckpt_folder = Path("logs/lightning_logs/version_0/checkpoints")
-    ckpt_file = ckpt_folder.joinpath("epoch=1-step=18000.ckpt")
+    ckpt_folder = Path("pls-logs/lightning_logs/version_4/checkpoints/")
+    ckpt_file = ckpt_folder.joinpath("epoch=1-step=147000.ckpt")
     assert ckpt_file.is_file()
     wrapper = Wrapper.load_from_checkpoint(checkpoint_path=ckpt_file)
     pt_file = ckpt_folder.joinpath("imagetext-base.pt")