pszemraj/t2t-re_pretrain-small
Viewer • Updated • 696k • 15
How to use pszemraj/ModernBERT2gpt2-700m-v0.1 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("pszemraj/ModernBERT2gpt2-700m-v0.1")
model = AutoModelForSeq2SeqLM.from_pretrained("pszemraj/ModernBERT2gpt2-700m-v0.1", device_map="auto")EncoderDecoder created from modernBERT-large and random-init gpt2 trained on the pszemraj/t2t-re_pretrain-small dataset for one epoch as a "baseline".
Logs and training script can be found on wandb
It achieves the following results on the evaluation set:
Base model
answerdotai/ModernBERT-large