Update README.md

README.md

@@ -46,7 +46,8 @@ We recommend using the following prompt template, which was used to fine-tune th
 <extra_id_1>Assistant\n
 ```
 
-Please note that a newline character `\n` should be added at the end of the prompt.
+- A newline character `\n` should be added at the end of the prompt.
+- `<extra_id_1>` should be used as a stop token.
 
 
 ## Usage
@@ -84,8 +85,7 @@ messages = [
     {"role": "user", "content": "Who are you?"},
 ]
 pipe = pipeline("text-generation", model="nvidia/Mistral-NeMo-Minitron-8B-Instruct")
-pipe.tokenizer = tokenizer  # You need to assign tokenizer manually
-pipe(messages)
+pipe(messages, max_length=64)
 ```
 
 ## AI Safety Efforts