Instructions to use allenai/PRIMERA-wcep with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use allenai/PRIMERA-wcep with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("allenai/PRIMERA-wcep") model = AutoModelForSeq2SeqLM.from_pretrained("allenai/PRIMERA-wcep") - 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=3.457e-05; Maximum converted output difference=3.457e-05.
cc @patrickvonplaten [HF maintainer(s) for this repo]
Ok to merge for me - thanks @joaogante
joaogante changed pull request status to merged