Update README.md

README.md

@@ -46,10 +46,10 @@ tokenizer = AutoTokenizer.from_pretrained("Zyphra/Zamba2-2.7B-instruct")
 model = AutoModelForCausalLM.from_pretrained("Zyphra/Zamba2-2.7B-instruct", device_map="cuda", torch_dtype=torch.bfloat16)
 
 # Format the input as a chat template
-input_string = "In one season a flower blooms three times. In one year, there is one blooming season. How many times do two flowers bloom in two years? Please include your logic."
-first_answer = "In one season, a flower blooms three times. In one year, there is one blooming season. Therefore, in two years, there are two blooming seasons. Since each flower blooms three times in one season, in two blooming seasons, each flower will bloom six times. Since there are two flowers, the total number of times they will bloom in two years is 12."
-second_input = "How many times do the two flowers blossom in three years?"
-sample = [{'role': 'user', 'content': input_string}, {'role': 'assistant', 'content': first_answer}, {'role': 'user', 'content': second_input}]
+user_turn_1 = "In one season a flower blooms three times. In one year, there is one blooming season. How many times do two flowers bloom in two years? Please include your logic."
+assistant_turn_1 = "In one season, a flower blooms three times. In one year, there is one blooming season. Therefore, in two years, there are two blooming seasons. Since each flower blooms three times in one season, in two blooming seasons, each flower will bloom six times. Since there are two flowers, the total number of times they will bloom in two years is 12."
+user_turn_2 = "How many times do the two flowers blossom in three years?"
+sample = [{'role': 'user', 'content': user_turn_1}, {'role': 'assistant', 'content': assistant_turn_1}, {'role': 'user', 'content': user_turn_2}]
 chat_sample = tokenizer.apply_chat_template(sample, tokenize=False)
 
 # Tokenize input and generate output