Update README.md

README.md

@@ -30,8 +30,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # Instantiate model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("Zyphra/Zamba2-7B-instruct")
-model = AutoModelForCausalLM.from_pretrained("Zyphra/Zamba2-7B-instruct", device_map="cuda", torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained("Zyphra/Zamba2-7B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("Zyphra/Zamba2-7B-Instruct", device_map="cuda", torch_dtype=torch.bfloat16)
 
 # Format the input as a chat template
 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."