Update README.md

README.md

@@ -15,7 +15,9 @@ model.eval()
 
 tokenizer = AutoTokenizer.from_pretrained(repo_id)
 text = "写一首春节主题的七言绝句"
-prompt = f"A chat between a curious human and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the human's questions.\n\n<|startofpiece|>{text}<|endofpiece|>"
+prompt = f"A chat between a curious human and an artificial intelligence assistant. "
+prompt += f"The assistant gives helpful, detailed, and polite answers to the human's questions.\n\n"
+prompt += f"<|startofpiece|>{text}<|endofpiece|>"
 tokens = tokenizer.encode_plus(prompt, truncation=False)["input_ids"]
 tokens = torch.tensor(tokens)[None,].to(device)
 with torch.no_grad():