metadata
license: apache-2.0
language:
- si
Fine tuned MT5 base model with Sinhala Wikipedia Dataset (Experimentally continues training)
This model is fine tuned with articles from Sinhala Wikipedia for article generation. Used around 10,000 articles for training and fine tuned arround 100 times.
How to use
We have to use "writeWiki: " part at the begining of each prompt.
You can use this model with a pipeline for text generation.
First you might need to install required libraries and import them.
!pip uninstall transformers -y
!pip install transformers
pip install tokenizers sentencepiece
Then we might need to restart the runtime either manually or use the below code to end it.
import os
os.kill(os.getpid(), 9)
Then we just have to import the tokenizer and run the pipeline:
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('google/mt5-small')
from transformers import pipeline
generator = pipeline(model='Suchinthana/MT5-Sinhala-Wikigen-Experimental', tokenizer=tokenizer)
generator("writeWiki: මානව ආහාර", do_sample=True, max_length=180)