Update README.md

README.md

@@ -69,10 +69,12 @@ We trained models using datasets of different sizes (200,000, 600,000, and 1,000
 |LR_scheduler | linear |
 
 ## Use model
-BELLE can be easily loaded with AutoModelForCausalLM.
-
-Please note that the input should be formatted as `Human: {input} \n\nAssistant:` in both training and inference.
+Please note that the input should be formatted as follows in both **training** and **inference**.
+``` python
+Human: {input} \n\nAssistant:
+``` 
 
+BELLE can be easily loaded with AutoModelForCausalLM.
 ``` python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import sys
@@ -146,6 +148,11 @@ BELLE模型以Bloomz-7b1-mt为基础，在 1.0M 条中文数据上，结合Stanf
 |LR_scheduler | linear |
 
 ## 使用模型
+请注意，在训练和推理时，模型的输入应该处理成如下形式：
+``` python
+Human: {input} \n\nAssistant:
+``` 
+
 通过AutoModelForCausalLM即可直接载入模型并使用。
 
 ``` python