readme: ---
  language: si
  tags:
  - Sinhala
  - text-generation
  - gpt2
  datasets:
  - mc4
  ---
  # Sinhala GPT2 trained on MC4 (manually cleaned)
  
  ### Overview
  
  This is a smaller GPT2 model trained on [MC4](https://github.com/allenai/allennlp/discussions/5056) Sinhala 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.
  
  This model uses a manually cleaned version of MC4 dataset which can be found [here](https://huggingface.co/datasets/keshan/clean-si-mc4). Although the dataset is relatively small ~3GB. The finetuned model on [news articles](https://huggingface.co/keshan/sinhala-gpt2-newswire) generates good and acceptable results. 
  
  ## Model Specification
  
  
  The model chosen for training is GPT2 with the following specifications:
   1. vocab_size=50257
   2. n_embd=768
   3. n_head=12
   4. n_layer=12
   5. n_positions=1024
   
  ## How to Use
  You can use this model directly with a pipeline for causal language modeling:
  
  ```py
  from transformers import pipeline
  generator = pipeline('text-generation', model='flax-community/Sinhala-gpt2')
  generator("මම", max_length=50, num_return_sequences=5)
  ```
  