Instructions to use sabre-code/pegasus-large-cnn-dailymail with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use sabre-code/pegasus-large-cnn-dailymail with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("sabre-code/pegasus-large-cnn-dailymail") model = AutoModelForSeq2SeqLM.from_pretrained("sabre-code/pegasus-large-cnn-dailymail", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "_from_model_config": true, | |
| "bos_token_id": 0, | |
| "decoder_start_token_id": 0, | |
| "eos_token_id": 1, | |
| "forced_eos_token_id": 1, | |
| "length_penalty": 0.8, | |
| "max_length": 256, | |
| "num_beams": 8, | |
| "pad_token_id": 0, | |
| "transformers_version": "4.34.1" | |
| } | |