Instructions to use allenai/PRIMERA-multinews with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use allenai/PRIMERA-multinews with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("allenai/PRIMERA-multinews") model = AutoModelForSeq2SeqLM.from_pretrained("allenai/PRIMERA-multinews", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Add TF weights
#2
by joaogante - opened
Model converted by the transformers' pt_to_tf CLI.
All converted model outputs and hidden layers were validated against its Pytorch counterpart. Maximum crossload output difference=2.989e-05; Maximum converted output difference=2.989e-05.
cc @patrickvonplaten [HF maintainer(s) for this repo]
Ok to merge for me - thanks @joaogante
joaogante changed pull request status to merged