Update README.md

README.md

@@ -12,11 +12,11 @@ pipeline_tag: reinforcement-learning
 
 Official release of TD-MPC2 model checkpoints for the paper
 
-[Scalable, Robust World Models for Continuous Control](https://www.tdmpc2.com) by
+[Scalable, Robust World Models for Continuous Control](https://nicklashansen.github.io/td-mpc2) by
 
 [Nicklas Hansen](https://nicklashansen.github.io), [Hao Su](https://cseweb.ucsd.edu/~haosu)\*, [Xiaolong Wang](https://xiaolonw.github.io)\* (UC San Diego)
 
-**Quick links:** [[Website]](https://www.tdmpc2.com) [[Paper]](https://www.tdmpc2.com) [[Dataset]](https://www.tdmpc2.com/dataset)
+**Quick links:** [[Website]](https://nicklashansen.github.io/td-mpc2) [[Paper]](https://arxiv.org/abs/2310.16828) [[Dataset]](https://www.tdmpc2.com/dataset)
 
 
 ## Model Details
@@ -33,7 +33,7 @@ We open-source a total of 324 TD-MPC2 model checkpoints, including 12 multi-task
 ### Model Sources
 
 - **Repository:** [https://github.com/nicklashansen/tdmpc2](https://github.com/nicklashansen/tdmpc2)
-- **Paper:** [https://www.tdmpc2.com](https://www.tdmpc2.com)
+- **Paper:** [https://www.tdmpc2.com](https://arxiv.org/abs/2310.16828)
 
 ## Uses
 
@@ -78,12 +78,13 @@ If you find our work useful, please consider citing the paper as follows:
 
 **BibTeX:**
 ```
-@article{Hansen2023TDMPC2,
-	title={TD-MPC2: Scalable, Robust World Models for Continuous Control},
-	author={Nicklas Hansen and Hao Su and Xiaolong Wang},
-	booktitle={arXiv preprint},
-	url={https://nicklashansen.github.io/td-mpc2},
-	year={2023}
+@misc{hansen2023tdmpc2,
+      title={TD-MPC2: Scalable, Robust World Models for Continuous Control}, 
+      author={Nicklas Hansen and Hao Su and Xiaolong Wang},
+      year={2023},
+      eprint={2310.16828},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG}
 }
 ```