Update README.md

README.md

@@ -94,15 +94,15 @@ text = "YOUR INPUT"
 inputs = tokenizer(text, return_tensors="pt").to(device)
 
 # Generate text
+# You can change the generation config
 generated_outputs = model.generate(
     input_ids=inputs['input_ids'],
     attention_mask=inputs['attention_mask'],
     max_length=1024,
     do_sample=True,
-    temperature=0.7,
-    top_p=0.95,
-    top_k=50,
-    repetition_penalty=1.2,
+    temperature=0.2,
+    top_p=0.9,
+    repetition_penalty=1.1,
     eos_token_id=tokenizer.eos_token_id,
 )