Update README.md

README.md

@@ -88,7 +88,7 @@ inputs = tokenizer(
 
 # generate text using the model
 streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
-outputs = model.generate(**inputs, max_length=2048, streamer=streamer, do_sample=True, eos_token_id=tokenizer.encode("<|im_end|>"))
+outputs = model.generate(**inputs, max_length=2048, streamer=streamer, do_sample=True, repetition_penalty=1.1, eos_token_id=tokenizer.encode("<|im_end|>"))
 
 # decode and print the output
 text = tokenizer.batch_decode(outputs)[0]