Update README.md

README.md

@@ -71,13 +71,9 @@ tokenizer.pad_token = tokenizer.eos_token
 
 ## inference
 prompt = """
-<|im_start|>system
-You are a helpful AI assistant.
-Take a deep breath and think step-by-step.<|im_end|>
+<s>[INST] Take a deep breath and think step-by-step.
 
-<|im_start|>user
-Write a tale about a time-traveling historian who's determined to witness the most significant events in human history.<|im_end|>
-<|im_start|>assistant
+Write a tale about a time-traveling historian who's determined to witness the most significant events in human history. [/INST]
 """
 
 # tokenize the input string