Update README.md

README.md

@@ -48,10 +48,10 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 messages = [
     {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
-    {"role": "user", "content": "Who are you?"},
+    {"role": "user", "content": "Who are you? Remember to respond in pirate speak!"},
 ]
 
-prompts = tokenizer.apply_chat_template(messages, tokenize=False)
+prompts = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
 llm = LLM(model=model_id)