Update README.md

README.md

@@ -45,9 +45,8 @@ end_token = "<|"
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2-large')
 model = GPT2LMHeadModel.from_pretrained('gpt2-large')
 tokenizer.add_special_tokens({'pad_token': '[PAD]'})
-tokenizer.add_special_tokens({'eos_token': '<|End|>'})
 special_tokens = {
-    "additional_special_tokens": ["<|USER|>", "<|SYSTEM|>", "<|ASSISTANT|>"]
+    "additional_special_tokens": ["<|USER|>", "<|ASSISTANT|>"]
 }
 tokenizer.add_special_tokens(special_tokens)
 model.resize_token_embeddings(len(tokenizer))