Instructions to use andreasmadsen/efficient_mlm_m0.40 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use andreasmadsen/efficient_mlm_m0.40 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="andreasmadsen/efficient_mlm_m0.40")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("andreasmadsen/efficient_mlm_m0.40") model = AutoModelForMaskedLM.from_pretrained("andreasmadsen/efficient_mlm_m0.40") - Notebooks
- Google Colab
- Kaggle
| inference: false | |
| This is a model checkpoint for ["Should You Mask 15% in Masked Language Modeling"](https://arxiv.org/abs/2202.08005) [(code)](https://github.com/princeton-nlp/DinkyTrain.git). | |
| The original checkpoint is avaliable at [princeton-nlp/efficient_mlm_m0.40](https://huggingface.co/princeton-nlp/efficient_mlm_m0.40). Unfortunately this checkpoint depends on code that isn't part of the official `transformers` | |
| library. Additionally, the checkpoints contains unused weights due to a bug. | |
| This checkpoint fixes the unused weights issue and uses the `RobertaPreLayerNorm` model from the `transformers` | |
| library. | |