UNIST-Eunchan
/

Research-Paper-Summarization-Pegasus-x-ArXiv

text2text-generation

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

UNIST-Eunchan commited on Nov 10, 2023

Commit

b79a664

•

1 Parent(s): 08748c3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -92,7 +92,7 @@ Paper Summarization
   ```(python)
   model.generate(input_ids =inputs["input_ids"].to(device),
                               attention_mask=inputs["attention_mask"].to(device),
-                              num_beam_groups=5,diversity_penalty=1.0,num_beams=5,min_length=100,max_length=128*4)
  ```

   ```(python)
   model.generate(input_ids =inputs["input_ids"].to(device),
                               attention_mask=inputs["attention_mask"].to(device),
+                              num_beam_groups=16,diversity_penalty=1.0,num_beams=16,min_length=100,max_length=128*4)
  ```