Update README.md

README.md

@@ -24,10 +24,10 @@ license: mit
 
 
 - **Developed by:** [Carnegie Mellon University Generative Intelligence Lab](https://www.cs.cmu.edu/~generative-intelligence-lab/)
-- **Funded by [optional]:** [More Information Needed]
+- **Funded by:** This work is partly supported by the Packard Foundation, Cisco Research Grant, and Amazon Faculty Award. This work is also in part supported by the Manufacturing Futures Institute, Carnegie Mellon University, through a grant from the Richard King Mellon Foundation. KD is supported by the Microsoft Research PhD Fellowship.
 - **Model type:** Autoregressive
 - **Language(s):** English
-- **License:** [More Information Needed]
+- **License:** MIT
 - **Finetuned from model:** [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct)
 
 ### Model Sources [optional]
@@ -35,8 +35,8 @@ license: mit
 <!-- Provide the basic links for the model. -->
 
 - **Repository:** [AvaLovelace1/LegoGPT](https://github.com/AvaLovelace1/LegoGPT)
-- **Paper [optional]:** TBA
-- **Demo [optional]:** TBA
+- **Paper:** [Generating Physically Stable and Buildable LEGO Designs from Text](https://huggingface.co/papers/2505.05469)
+- **Demo:** TBA
 
 ## Uses
 
@@ -132,15 +132,16 @@ The model was fine-tuned using LoRA applied to the `q_proj` and `v_proj` matrice
 
 ## Citation
 
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+If you find this model useful for your research, please cite the following work.
 
-**BibTeX:**
-
-TBA
-
-**APA:**
-
-TBA
+```bibtex
+@article{pun2025legogpt,
+    title   = {Generating Physically Stable and Buildable LEGO Designs from Text},
+    author  = {Pun, Ava and Deng, Kangle and Liu, Ruixuan and Ramanan, Deva and Liu, Changliu and Zhu, Jun-Yan},
+    journal = {arXiv preprint arXiv:2505.05469},
+    year    = {2025}
+}
+```
 
 ## Model Card Contact