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