How to use this model directly from the
from transformers import AutoTokenizer, AutoModelWithLMHead tokenizer = AutoTokenizer.from_pretrained("lvwerra/gpt2-medium-taboo") model = AutoModelWithLMHead.from_pretrained("lvwerra/gpt2-medium-taboo")
A fine-tuned GPT-2 version for Taboo cards generation.
The model was trained on ~900 Taboo cards in the following format for 100 epochs:
Describe the word Glitch without using the words Problem, Unexpected, Technology, Minor, Outage.