PartiallyTyped's picture
Update README.md
508b164
metadata
language:
  - fi
tags:
  - text generation
license: mit
datasets:
  - answerable tydiqa

ReadMe

This is a pretrained model based on Finnish-NLP/gpt2-finnish that has been trained on copenlu/answerable_tydiqa, specifically the text field of the Finnish samples for 2 epochs.

To use the pretrained head, use: AutoModelForCausalLM.from_pretrained.

from transformers import AutoModelForCausalLM, AutoTokenizer

model_path = "PartiallyTyped/answerable_tydiqa_lm_pretrained_finnish"
model = AutoModelForCausalLM.from_pretrained(model_path)
tokenizer = AutoTokenizer.from_pretrained(model_path)