Increasing The Performance of Cognitively Inspired Data-Efficient Language Models via Implicit Structure Building
Paper • 2310.20589 • Published
# Load model directly
from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained("omarmomen/transformer_base_final_2", trust_remote_code=True, dtype="auto")This model is part of the experiments in the published paper at the BabyLM workshop in CoNLL 2023. The paper titled "Increasing The Performance of Cognitively Inspired Data-Efficient Language Models via Implicit Structure Building" (https://aclanthology.org/2023.conll-babylm.29/)
omarmomen/transformer_base_final_2 is a baseline vanilla transformer encoder.
The model is pretrained on the BabyLM 10M dataset using a custom pretrained RobertaTokenizer (https://huggingface.co/omarmomen/babylm_tokenizer_32k).
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="omarmomen/transformer_base_final_2", trust_remote_code=True)