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

510a7fc

•

1 Parent(s): b79a664

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -93,8 +93,7 @@ Paper Summarization
   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)
- ```

   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)
+  ```