Instructions to use vocab-transformers/distilbert-tokenizer_256k-MLM_500k with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use vocab-transformers/distilbert-tokenizer_256k-MLM_500k with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="vocab-transformers/distilbert-tokenizer_256k-MLM_500k")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("vocab-transformers/distilbert-tokenizer_256k-MLM_500k") model = AutoModelForMaskedLM.from_pretrained("vocab-transformers/distilbert-tokenizer_256k-MLM_500k") - Notebooks
- Google Colab
- Kaggle
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
DistilBERT with 256k token embeddings
This model was initialized with a word2vec token embedding matrix with 256k entries, but these token embeddings were updated during MLM. The word2vec was trained on 100GB data from C4, MSMARCO, News, Wikipedia, S2ORC, for 3 epochs.
Then the model was trained on this dataset with MLM for 500k steps (batch size 64). The token embeddings were updated during MLM.
For the same model but with frozen token embeddings while MLM training see: https://huggingface.co/vocab-transformers/distilbert-word2vec_256k-MLM_500k
- Downloads last month
- 8