Update README.md

README.md

@@ -40,7 +40,14 @@ model = AutoModel.from_pretrained(
 
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True, use_fast=False)
 
-generation_config = GenerationConfig(max_new_tokens=2048, do_sample=True, temperature=0.01, top_p=0.01, repetition_penalty=1.1)
+generation_config = GenerationConfig(
+  max_new_tokens=2048,
+  do_sample=True,
+  temperature=0.01,
+  top_p=0.01,
+  repetition_penalty=1.1,
+  eos_token_id=128009
+)
 
 prompt_engine = MRPromptV3()