Update README.md

README.md

@@ -65,7 +65,7 @@ from transformers import (
 ```
 
 ```python
-peft_model_id = "dfurman/Yi-6B-instruct-v0.1"
+peft_model_id = "dfurman/Mistral-7B-Instruct-v0.1"
 config = PeftConfig.from_pretrained(peft_model_id)
 
 tokenizer = AutoTokenizer.from_pretrained(
@@ -95,7 +95,6 @@ model = PeftModel.from_pretrained(
 
 ```python
 messages = [
-    {"role": "system", "content": "You are a helpful assistant. Respond as briefly as possible."},    
     {"role": "user", "content": "Tell me a recipe for a mai tai."},
 ]
 
@@ -135,22 +134,12 @@ print(response)
 
 **Prompt**: 
 ```python
-"""<|im_start|>system
-You are a helpful assistant. Respond as briefly as possible.<|im_end|>
-<|im_start|>user
-Tell me a recipe for a mai tai.<|im_end|>
-<|im_start|>assistant"""
+coming
 ```
 
 **Generation**:
 ```python
-"""Here's one simple version of the classic Mai Tai cocktail:
-
-1 oz White Rum (Bacardi, Don Papa, etc.) ➕ ½ oz Coconut Cream Liqueur (Malibu or Coco Lopez)
-2 tsp Simple Syrup ➕ Dash Orange Bitters
-3-4 Ice Cubes
-
-Shake all ingredients in a shaker filled with ice until well chilled and strain into an old fashioned glass over fresh crushed ice. Garnish with mint leaves if desired. Enjoy!"""
+coming
 ```
 
 </details>
@@ -192,7 +181,7 @@ print(prompt)
 <summary>Output</summary>
 
 ```python
-"<s>[INST] What is your favourite condiment? [/INST]Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!</s> [INST] Do you have mayonnaise recipes? [/INST]"
+coming
 ```
 </details>