legacy-datasets/wikipedia
Updated • 125k • 630
How to use Suchinthana/T5-Base-Wikigen with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("Suchinthana/T5-Base-Wikigen")
model = AutoModelForSeq2SeqLM.from_pretrained("Suchinthana/T5-Base-Wikigen")This model is fine tuned with articles from Simple English Wikipedia for article generation. Used around 25,000 articles for training.
We have to use "writeWiki: " part at the begining of each prompt.
You can use this model directly with a pipeline for text generation. This example generates a different sequence each time it's run:
>>> from transformers import pipeline
>>> generator = pipeline('text2text-generation', model='Suchinthana/T5-Base-Wikigen')
>>> generator("writeWiki: Microcontroller", do_sample=True, max_length=250)