Update README.md
Browse files
README.md
CHANGED
|
@@ -51,6 +51,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 51 |
torch_dtype=torch.bfloat16)
|
| 52 |
|
| 53 |
messages = [
|
|
|
|
| 54 |
{"role": "user", "content": "明朝最后一位皇帝是谁？回答他的名字，然后停止输出"},
|
| 55 |
]
|
| 56 |
prompt = tokenizer.apply_chat_template(
|
|
|
|
| 51 |
torch_dtype=torch.bfloat16)
|
| 52 |
|
| 53 |
messages = [
|
| 54 |
+
{"role": "system", "content": "你是一个AI助手。"},
|
| 55 |
{"role": "user", "content": "明朝最后一位皇帝是谁？回答他的名字，然后停止输出"},
|
| 56 |
]
|
| 57 |
prompt = tokenizer.apply_chat_template(
|