Update README.md

README.md

@@ -65,15 +65,15 @@ model= AutoModelForCausalLM.from_pretrained("tuanle/VN-News-GPT2").to(device)
 input_ids = tokenizer.encode(text, return_tensors='pt').to(device)
 sample_outputs = model.generate(input_ids,
                                 do_sample=True,
-                                max_length=max_len,
-                                min_length=min_len,
+                                max_length= 768,
+                                min_length= 60,
                                 #    temperature = .8,
-                                top_k= top_k,
-                                top_p = top_p,
-                                num_beams= num_beams,
+                                top_k= 100,
+                                top_p = 0.7,
+                                num_beams= 5,
                                 early_stopping= True,
                                 no_repeat_ngram_size= 2  ,
-                                num_return_sequences= num_return_sequences)
+                                num_return_sequences= 3)
 
 for i, sample_output in enumerate(sample_outputs):
     temp = tokenizer.decode(sample_output.tolist())