Downloads · 30 days
34
2% of all-time downloads
Sinhala-gpt2
flax-community/Sinhala-gpt2
Sinhala-gpt2 is a text generation model from flax-community. Use it when you need the model to write or continue text. It is set up for transformers.
This is a smaller GPT2 model trained on MC4 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 mo…
All-time downloads
1.9K
Public
Repo size
20.2 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 33%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 19, 2021
- SHA
- ea266fac
Datasets
Model card
From the Hugging Face model README
Sinhala GPT2 trained on MC4 (manually cleaned)
Overview
This is a smaller GPT2 model trained on MC4 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. Although the dataset is relatively small ~3GB. The finetuned model on news articles generates good and acceptable results.
Model Specification
The model chosen for training is GPT2 with the following specifications:
- vocab_size=50257
- n_embd=768
- n_head=12
- n_layer=12
- n_positions=1024
How to Use
You can use this model directly with a pipeline for causal language modeling:
from transformers import pipeline
generator = pipeline('text-generation', model='flax-community/Sinhala-gpt2')
generator("මම", max_length=50, num_return_sequences=5)
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Languages
- si
- Created
- Mar 2, 2022
- Updated
- Jul 19, 2021