Update README.md

README.md

@@ -9,4 +9,17 @@ Pretrained models for Make-An-Agent. See https://github.com/cheryyunl/Make-An-Ag
 
 `autoencoder.pt` and `behavior_embedding.pt` are trained with parameter data and trajectory data in `train_data/data.pt` of our dataset.
 
-`model-best.pt` is the policy generator model trained by latent parameter representation data and behavior embedding data in `train_data/process_data.pt`
+`model-best.pt` is the policy generator model trained by latent parameter representation data and behavior embedding data in `train_data/process_data.pt`
+
+## 📝 Citation
+
+If you find our model or dataset useful, please consider citing as follows:
+
+```
+@article{liang2024make,
+title={Make-An-Agent: A Generalizable Policy Network Generator with Behavior-Prompted Diffusion},
+author={Liang, Yongyuan and Xu, Tingqiang and Hu, Kaizhe and Jiang, Guangqi and Huang, Furong and Xu, Huazhe},
+journal={arXiv preprint arXiv:2407.10973},
+year={2024}
+}
+```