Update README.md

README.md

@@ -82,17 +82,18 @@ Paper Summarization
                               length_penalty=1, num_beams=2, max_length=128*4,min_length=150, no_repeat_ngram_size= 3, top_k=25,top_p=0.95)
     
   ```
-  - R-1 | R-2 | R-L | R-LSUM : 40.8486 | 16.3717 | 25.2937 | 33.6923 at 
+  - R-1 | R-2 | R-L | R-LSUM : 40.8486 | 16.3717 | 25.2937 | 33.6923 (refer to **PEGASUS-X's [paper](https://arxiv.org/pdf/2208.04347.pdf)**) at 
   ```(python)
   model.generate(input_ids =inputs["input_ids"].to(device),
                               attention_mask=inputs["attention_mask"].to(device),
                               length_penalty=1, num_beams=1, max_length=128*2,top_p=1)
   ```
-  - R-1 | R-2 | R-L | R-LSUM : TBD | TBD | TBD | TBD at 
+  - R-1 | R-2 | R-L | R-LSUM : TBD | TBD | TBD | TBD (**Diverse Beam-Search Decoding**) at 
   ```(python)
   model.generate(input_ids =inputs["input_ids"].to(device),
                               attention_mask=inputs["attention_mask"].to(device),
-                              num_beam_groups=16,num_beams=16,min_length=100,max_length=128*4)
+                              num_beam_groups=5,diversity_penalty=1.0,num_beams=5,min_length=100,max_length=128*4)
+    
  ```