Summarization
Transformers
PyTorch
Safetensors
English
bart
text2text-generation
booksum
summary
book
Instructions to use KamilAin/bart-base-booksum with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use KamilAin/bart-base-booksum 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="KamilAin/bart-base-booksum", device_map="auto")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("KamilAin/bart-base-booksum") model = AutoModelForSeq2SeqLM.from_pretrained("KamilAin/bart-base-booksum", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| {"unk_token": "<unk>", "bos_token": "<s>", "eos_token": "</s>", "add_prefix_space": false, "errors": "replace", "sep_token": "</s>", "cls_token": "<s>", "pad_token": "<pad>", "mask_token": "<mask>", "model_max_length": 1024, "special_tokens_map_file": null, "name_or_path": "facebook/bart-base", "tokenizer_class": "BartTokenizer"} |