Update README.md

README.md

@@ -67,7 +67,7 @@ export FLUX2_QWEN3_MODEL_SPEC=Qwen/Qwen3-4B
 Evaluate on LIBERO:
 
 ```bash
-export CKPT_PATH="$(pwd)/checkpoints/imagewam_release/libero/flux2_klein_4b/checkpoint.pt"
+export CKPT_PATH="$(pwd)/checkpoints/imagewam_release/libero/flux2_klein_4b/model.pt"
 export DATASET_STATS_PATH="$(pwd)/checkpoints/imagewam_release/libero/flux2_klein_4b/dataset_stats.json"
 
 NUM_GPUS=8 FLUX2_VARIANT=4b bash scripts/flux2/run_eval_flux2_libero.sh
@@ -95,11 +95,14 @@ This checkpoint is not intended for safety-critical or real-world robot deployme
 If you use this checkpoint, please cite the ImageWAM paper:
 
 ```bibtex
-@misc{imagewam2026,
-  title  = {ImageWAM: Do World Action Models Really Need Video Generation, or Just Image Editing?},
-  author = {TODO},
-  year   = {2026},
-  note   = {TODO: add arXiv or conference information}
+@misc{zhang2026imagewam,
+      title={ImageWAM: Do World Action Models Really Need Video Generation, or Just Image Editing?}, 
+      author={Yuyang Zhang and Wenyao Zhang and Zekun Qi and He Zhang and Haitao Lin and Jingbo Zhang and Yao Mu and Xiaokang Yang and Wenjun Zeng and Xin Jin},
+      year={2026},
+      eprint={2606.19531},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2606.19531}, 
 }
 ```