Update README.md

README.md

@@ -1,5 +1,5 @@
 ---
-license: mit
+license: apache-2.0
 datasets:
 - liuhaotian/LLaVA-Pretrain
 - liuhaotian/LLaVA-Instruct-150K
@@ -85,3 +85,18 @@ print(processor.decode(output[0][2:], skip_special_tokens=True))
 
 ## Contact
 This model was trained by [Baichuan Zhou](https://baichuanzhou.github.io/), from Beihang Univerisity, under the supervision of  [Prof. Lei Huang](https://huangleibuaa.github.io/).
+
+## ✏ Citation
+
+If you find our paper and code useful in your research, please consider giving a star :star: and citation :pencil:.
+
+```BibTeX
+@misc{zhou2024tinyllava,
+      title={TinyLLaVA: A Framework of Small-scale Large Multimodal Models}, 
+      author={Baichuan Zhou and Ying Hu and Xi Weng and Junlong Jia and Jie Luo and Xien Liu and Ji Wu and Lei Huang},
+      year={2024},
+      eprint={2402.14289},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG}
+}
+```