Edit model card

Model Card for honicky/t5-short-story-character-extractor

I trained this model as part of a learning project to build a children's story authoring tool for parents of young children. See http://www.storytime.glass/ This model takes in a short story and outputs a comma separated list of characters in the story.

I'm not sure yet how useful this fine-tune is: rather it is for me to learn about the nuts and bolts of fine-tuning.

Model Details

The model is a fine-tune of a sequence-to-sequence model Flan T5 Large, so a different architecture from decoder-only models like GPT*. Maybe this allows it to perform this transformation task (transform a story into a list of characters) using a smaller model?

  • Trained using transformers.Seq2SeqTrainer plus the corresponding collator, tokenizer etc

  • Developed by: RJ Honicky
  • Model type: Encoder-Decoder Transformer
  • Language(s): English (fine tune data set)
  • License: MIT
  • Finetuned from model: google/flan-t5-large

Model Sources


Primarily for use in https://github.com/honicky/story-time and for learning.

Downloads last month
Unable to determine this model’s pipeline type. Check the docs .

Adapter for