README.md

metadata

language: it
tags:
  - adaption
  - recycled
  - gpt2-small
pipeline_tag: text-generation

GPT-2 recycled for Italian (small)

Wietse de Vries • Malvina Nissim

Model description

This model is based on the small OpenAI GPT-2 (gpt2) model.

For details, check out our paper on arXiv and the code on Github.

Related models

Dutch

• gpt2-small-dutch-embeddings: Small model size with only retrained lexical embeddings.
• gpt2-small-dutch: Small model size with retrained lexical embeddings and additional fine-tuning of the full model. (Recommended)
• gpt2-medium-dutch-embeddings: Medium model size with only retrained lexical embeddings.

Italian

• gpt2-small-italian-embeddings: Small model size with only retrained lexical embeddings.
• gpt2-small-italian: Small model size with retrained lexical embeddings and additional fine-tuning of the full model. (Recommended)
• gpt2-medium-italian-embeddings: Medium model size with only retrained lexical embeddings.

How to use

from transformers import pipeline

pipe = pipeline("text-generation", model="GroNLP/gpt2-small-italian")

from transformers import AutoTokenizer, AutoModel, TFAutoModel

tokenizer = AutoTokenizer.from_pretrained("GroNLP/gpt2-small-italian")
model = AutoModel.from_pretrained("GroNLP/gpt2-small-italian")  # PyTorch
model = TFAutoModel.from_pretrained("GroNLP/gpt2-small-italian")  # Tensorflow

BibTeX entry

@misc{devries2020good,
      title={As good as new. How to successfully recycle English GPT-2 to make models for other languages}, 
      author={Wietse de Vries and Malvina Nissim},
      year={2020},
      eprint={2012.05628},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}