Update README.md

README.md

@@ -4,5 +4,32 @@ license: afl-3.0
 
 
 ## Model description
-'MathGLM-10B is finetuned from GLM-10B on a dataset with additional multi-step arithmetic operations and math problems described in text, achieves similar performance to GPT-4 on a 5,000-samples Chinese math problem
-test set.'
+MathGLM-10B is finetuned from GLM-10B on a dataset with additional multi-step arithmetic operations and math problems described in text, achieves similar performance to GPT-4 on a 5,000-samples Chinese math problem
+test set.
+
+
+## How to use 
+```python
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("BAAI/glm-10b-chinese", trust_remote_code=True)
+model = AutoModelForSeq2SeqLM.from_pretrained("BAAI/glm-10b-chinese", trust_remote_code=True)
+model = model.half().cuda()
+
+inputs = tokenizer("凯旋门位于意大利米兰市古城堡旁。1807年为纪念[MASK]而建，门高25米，顶上矗立两武士青铜古兵车铸像。", return_tensors="pt")
+inputs = tokenizer.build_inputs_for_generation(inputs, max_gen_length=512)
+inputs = {key: value.cuda() for key, value in inputs.items()}
+outputs = model.generate(**inputs, max_length=512, eos_token_id=tokenizer.eop_token_id)
+print(tokenizer.decode(outputs[0].tolist()))
+```
+
+## Citation
+Please cite our paper if you find this code useful for your research:
+```
+@article{yang2023gpt,
+  title={GPT Can Solve Mathematical Problems Without a Calculator},
+  author={Yang, Zhen and Ding, Ming and Lv, Qingsong and Jiang, Zhihuan and He, Zehai and Guo, Yuyi and Bai, Jinfeng and Tang, Jie},
+  journal={arXiv preprint arXiv:2309.03241},
+  year={2023}
+}
+```
+