Update README.md

updated generation settings

README.md

@@ -124,7 +124,7 @@ unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
 
 _, reft_response = reft_model.generate(
     prompt, unit_locations={"sources->base": (None, unit_locations)},
-    intervene_on_prompt=True, max_new_tokens=216, do_sample=True, top_k=50,temperature=0.7,
+    intervene_on_prompt=True, max_new_tokens=537, do_sample=True, top_k=50,temperature=0.7,
     eos_token_id=terminators, early_stopping=True
 )
 print(tokenizer.decode(reft_response[0], skip_special_tokens=True))