keshan/SinhalaBERTo

Overview

This is a slightly smaller model trained on OSCAR Sinhala dedup dataset. As Sinhala is one of those low resource languages, there are only a handful of models been trained. So, this would be a great place to start training for more downstream tasks.

Model Specification

The model chosen for training is Roberta with the following specifications:

1. 1.vocab_size=52000
2. 2.maxpositionembeddings=514
3. 3.numattentionheads=12
4. 4.numhiddenlayers=6
5. 5.typevocabsize=1

How to Use

You can use this model directly with a pipeline for masked language modeling:

from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline

model = AutoModelWithLMHead.from_pretrained("keshan/SinhalaBERTo")
tokenizer = AutoTokenizer.from_pretrained("keshan/SinhalaBERTo")

fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)

fill_mask("මම ගෙදර <mask>.")