Update README.md

README.md

@@ -21,7 +21,7 @@ python3 embedding_convert.py \
     --meta_llama_pth_file /path_to_LLaMA/llama-7b/consolidated.00.pth
 ```
 
-模型权重还原后，可通过以下代码调试运行：
+BiLLa-7B-SFT模型权重还原后，可通过以下代码调试运行：
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -42,3 +42,9 @@ output_ids = output_ids[0][len(input_ids[0]):]
 outputs = tokenizer.decode(output_ids, skip_special_tokens=True).strip()
 print(outputs)
 ```
+
+BiLLa-7B-SFT的模型输入需按以下格式构造（注意`Assistant:`后必须有一个空格）：
+```
+Human: [Your question]
+Assistant: 
+```