Update README.md
Browse files
README.md
CHANGED
|
@@ -20,7 +20,7 @@ device = "cuda" # for GPU usage or "cpu" for CPU usage
|
|
| 20 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 21 |
model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
|
| 22 |
|
| 23 |
-
input_content = "
|
| 24 |
|
| 25 |
messages = [{"role": "user", "content": input_content}]
|
| 26 |
input_text=tokenizer.apply_chat_template(messages, tokenize=False)
|
|
|
|
| 20 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 21 |
model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
|
| 22 |
|
| 23 |
+
input_content = "你是一个文本纠错专家，纠正输入句子中的语法错误，并输出正确的句子，输入句子为：\n少先队员因该为老人让坐。"
|
| 24 |
|
| 25 |
messages = [{"role": "user", "content": input_content}]
|
| 26 |
input_text=tokenizer.apply_chat_template(messages, tokenize=False)
|