Update README.md

README.md

@@ -110,15 +110,9 @@ print(tokenizer.decode(input_ids[0]))
 print("\n\n*** Generate:")
 with torch.autocast("cuda", dtype=torch.bfloat16):
     output = model.generate(
-        input_ids=input_ids.cuda(),
+        input_ids=input_ids.to("cuda"),
         max_new_tokens=1024,
-        do_sample=True,
-        temperature=0.7,
         return_dict_in_generate=True,
-        eos_token_id=tokenizer.eos_token_id,
-        pad_token_id=tokenizer.pad_token_id,
-        repetition_penalty=1.2,
-        no_repeat_ngram_size=5,
     )
 
 response = tokenizer.decode(
@@ -128,20 +122,16 @@ response = tokenizer.decode(
 print(response)
 ```
 
-<details>
-
-<summary>Outputs</summary>
-
-**Prompt**:
+**Outputs**
 
 ```python
-"<s> [INST] Tell me a recipe for a mai tai. [/INST]"
-```
+*** **Prompt**:
 
-**Generation**:
+<s> [INST] Tell me a recipe for a mai tai. [/INST]
 
-```python
-"""1.5 oz White Rum
+*** **Generation**
+
+1.5 oz White Rum
 2 oz Dark Rum
 1 oz Orange Curacao
 0.5 oz Orgeat Syrup
@@ -152,11 +142,9 @@ In a shaker filled with ice, combine the white rum, dark rum, orange curacao, or
 
 Strain the mixture into a double old-fashioned glass filled with fresh ice. Garnish with a lime wedge and a sprig of mint.
 
-Enjoy your delicious mai tai!"""
+Enjoy your delicious mai tai!
 ```
 
-</details>
-
 
 ## Speeds, Sizes, Times