princeton-nlp
/

Sheared-LLaMA-1.3B-ShareGPT

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

princeton-nlp commited on Dec 4, 2023

Commit

d2f3cfa

•

1 Parent(s): 43c5122

Update README.md

Files changed (1) hide show

README.md +4 -3

README.md CHANGED Viewed

@@ -25,9 +25,10 @@ model = LlamaModelForCausalLM.from_pretrained("princeton-nlp/Sheared-LLaMA-1.3B-
 If you find our model useful, consider citing us with:
 ```
 @article{xia2023sheared,
-   title={Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning},
-   author={Xia, Mengzhou and Gao, Tianyu, and Zeng, Zhiyuan and Chen, Danqi},
-   year={2023}
 }
 ```

 If you find our model useful, consider citing us with:
 ```
 @article{xia2023sheared,
+  title={Sheared llama: Accelerating language model pre-training via structured pruning},
+  author={Xia, Mengzhou and Gao, Tianyu and Zeng, Zhiyuan and Chen, Danqi},
+  journal={arXiv preprint arXiv:2310.06694},
+  year={2023}
 }
 ```