omarmomen/babylm_10M
Viewer • Updated • 1.21M • 173
How to use omarmomen/tf_babylm_1 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="omarmomen/tf_babylm_1", trust_remote_code=True) # Load model directly
from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained("omarmomen/tf_babylm_1", trust_remote_code=True, dtype="auto")This model is part of the experiments in my master's thesis titled "Linguistic Structure Induction from Language Models" (https://arxiv.org/abs/2403.09714).
"omarmomen/tf_babylm_1" is the vanilla transformer encoder referred to in Chapter 5 (p. 59).
The model is trained on the BabyLM 10M dataset, with a RobertaTokenizer pretrained on the BabyLM 10M dataset with 16K tokens (https://huggingface.co/omarmomen/babylm_bpe_tokenizer_16k).