Update README.md (#6)

- Update README.md (34f8ba9062ad5b4157f7f764a0f368a310e24fcb)

README.md

@@ -83,11 +83,11 @@ messages = [
     {"role": "user", "content": "Who are you?"},
 ]
 
+# this should work perfectly for the model to stop generating
 terminators = [
-    tokenizer.eos_token_id,
-    tokenizer.convert_tokens_to_ids("<|im_end|>"),
-    tokenizer.convert_tokens_to_ids("<|assistant|>"),
-    tokenizer.convert_tokens_to_ids("<|end|>")
+    tokenizer.eos_token_id, # this should be <|im_end|>
+    tokenizer.convert_tokens_to_ids("<|assistant|>"), # sometimes model stops generating at <|assistant|>
+    tokenizer.convert_tokens_to_ids("<|end|>") # sometimes model stops generating at <|end|>
 ]
 
 pipe = pipeline(