Instructions to use allenai/PRIMERA-multixscience with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use allenai/PRIMERA-multixscience with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("allenai/PRIMERA-multixscience") model = AutoModelForSeq2SeqLM.from_pretrained("allenai/PRIMERA-multixscience") - Notebooks
- Google Colab
- Kaggle
Add default num_beams size to config
#2
by johngiorgi - opened
This model uses num_beams=5 during decoding to get the results reported in the paper. I have added it to the config, so that HuggingFace scripts will use this beam size by default.
johngiorgi changed pull request status to merged