Update README.md

README.md

@@ -65,7 +65,7 @@ model.to(device)
 inputs = encodeds.to(device)
 
 # Increase max_new_tokens if needed
-generated_ids = model.generate(inputs, max_new_tokens=256,temperature=0.1, top_p=0.90, do_sample=True,pad_token_id=50259,eos_token_id=50259,num_return_sequences=1)
+generated_ids = model.generate(inputs, max_new_tokens=256,temperature=0.1, top_p=0.90, do_sample=True,pad_token_id=50258,eos_token_id=50258,num_return_sequences=1)
 print(tokenizer.decode(generated_ids[0]).replace(prompt,'').split('<eos>')[0])
 e = time.time()
 print(f'time taken:{e-s}')