Instructions to use lucadiliello/bart-small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use lucadiliello/bart-small with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("lucadiliello/bart-small") model = AutoModelForSeq2SeqLM.from_pretrained("lucadiliello/bart-small", device_map="auto") - Notebooks
- Google Colab
- Kaggle
bart-small is a lighter version of bart-base with less attention heads, smaller FFT and smaller hidden-size.
More details can be found here.
Citation
@software{Di_Liello_bart-small,
author = {Di Liello, Luca},
license = {GPL-2.0},
title = {{bart-small}},
url = {https://github.com/lucadiliello/bart-small}
}