Update README.md

README.md

@@ -16,7 +16,7 @@ model-index:
 
 # Mayonnaise LLM
 
-Mayo is a language model fine-tuned on the [Mayo dataset](https://huggingface.co/datasets/nroggendorff/mayo) using Supervised Fine-Tuning (SFT) and Teacher Reinforced Learning (TRL) techniques. It is based on the [TinyLlama/TinyLlama-1.1B-Chat-v1.0 model](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0).
+Mayo is a language model fine-tuned on the [Mayo dataset](https://huggingface.co/datasets/nroggendorff/mayo) using Supervised Fine-Tuning (SFT) and Teacher Reinforced Learning (TRL) techniques. It is based on the [Mistral 7b model](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3)
 
 ## Features
 
@@ -56,7 +56,7 @@ model_id = "nroggendorff/mayo"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
 
-prompt = "<|user|>\nWhat color is the sky?</s>\n"
+prompt = "[INST] What color is the sky? [/INST]"
 inputs = tokenizer(prompt, return_tensors="pt")
 
 outputs = model.generate(**inputs, max_new_tokens=32)