Downloads · 30 days
363K
1% of all-time downloads
distilbart-cnn-12-6
sshleifer/distilbart-cnn-12-6
distilbart-cnn-12-6 is a summarization model from sshleifer. Use it when you need a shorter version of a longer text. It is set up for transformers. The card lists the license as apache-2.0.
This checkpoint should be loaded into BartForConditionalGeneration.frompretrained. See the BART docs for more information.
All-time downloads
46.7M
Public
Repo size
19.9 GB
Likes
326
Public
Repo makeup
Click a slice to open those files.
.ot1.6 GB · 40%
At a glance
- Task
- Summarization
- Library
- transformers
- License
- apache-2.0
- Model type
- bart
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jun 14, 2021
- SHA
- a4f8f3ea
Try a prompt
Datasets
Model card
From the Hugging Face model README
Usage
This checkpoint should be loaded into BartForConditionalGeneration.from_pretrained. See the BART docs for more information.
Metrics for DistilBART models
| Model Name | MM Params | Inference Time (MS) | Speedup | Rouge 2 | Rouge-L |
|---|---|---|---|---|---|
| distilbart-xsum-12-1 | 222 | 90 | 2.54 | 18.31 | 33.37 |
| distilbart-xsum-6-6 | 230 | 132 | 1.73 | 20.92 | 35.73 |
| distilbart-xsum-12-3 | 255 | 106 | 2.16 | 21.37 | 36.39 |
| distilbart-xsum-9-6 | 268 | 136 | 1.68 | 21.72 | 36.61 |
| bart-large-xsum (baseline) | 406 | 229 | 1 | 21.85 | 36.50 |
| distilbart-xsum-12-6 | 306 | 137 | 1.68 | 22.12 | 36.99 |
| bart-large-cnn (baseline) | 406 | 381 | 1 | 21.06 | 30.63 |
| distilbart-12-3-cnn | 255 | 214 | 1.78 | 20.57 | 30.00 |
| distilbart-12-6-cnn | 306 | 307 | 1.24 | 21.26 | 30.59 |
| distilbart-6-6-cnn | 230 | 182 | 2.09 | 20.17 | 29.70 |
- Task
- Summarization
- Library
- transformers
- Type
- bart
- License
- apache-2.0
- Languages
- en
- Hf Inference
- live
- Created
- Mar 2, 2022
- Updated
- Jun 14, 2021