Summarization
Transformers
PyTorch
TensorFlow
JAX
Rust
Safetensors
English
bart
text2text-generation
Eval Results (legacy)
Instructions to use facebook/bart-large-cnn with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use facebook/bart-large-cnn with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("summarization", model="facebook/bart-large-cnn")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn") model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +2 -4
config.json
CHANGED
|
@@ -10,6 +10,7 @@
|
|
| 10 |
"decoder_ffn_dim": 4096,
|
| 11 |
"decoder_layerdrop": 0.0,
|
| 12 |
"decoder_layers": 12,
|
|
|
|
| 13 |
"do_sample": false,
|
| 14 |
"dropout": 0.1,
|
| 15 |
"early_stopping": true,
|
|
@@ -17,9 +18,6 @@
|
|
| 17 |
"encoder_ffn_dim": 4096,
|
| 18 |
"encoder_layerdrop": 0.0,
|
| 19 |
"encoder_layers": 12,
|
| 20 |
-
"eos_token_ids": [
|
| 21 |
-
2
|
| 22 |
-
],
|
| 23 |
"finetuning_task": null,
|
| 24 |
"id2label": {
|
| 25 |
"0": "LABEL_0",
|
|
@@ -55,4 +53,4 @@
|
|
| 55 |
"torchscript": false,
|
| 56 |
"use_bfloat16": false,
|
| 57 |
"vocab_size": 50264
|
| 58 |
-
}
|
|
|
|
| 10 |
"decoder_ffn_dim": 4096,
|
| 11 |
"decoder_layerdrop": 0.0,
|
| 12 |
"decoder_layers": 12,
|
| 13 |
+
"decoder_start_token_id": 2,
|
| 14 |
"do_sample": false,
|
| 15 |
"dropout": 0.1,
|
| 16 |
"early_stopping": true,
|
|
|
|
| 18 |
"encoder_ffn_dim": 4096,
|
| 19 |
"encoder_layerdrop": 0.0,
|
| 20 |
"encoder_layers": 12,
|
|
|
|
|
|
|
|
|
|
| 21 |
"finetuning_task": null,
|
| 22 |
"id2label": {
|
| 23 |
"0": "LABEL_0",
|
|
|
|
| 53 |
"torchscript": false,
|
| 54 |
"use_bfloat16": false,
|
| 55 |
"vocab_size": 50264
|
| 56 |
+
}
|