Language Generation Minimal Start

1
2
3
4
5
6
7
8
9
10
11
12
import logging

from simpletransformers.language_generation import LanguageGenerationModel, LanguageGenerationArgs


logging.basicConfig(level=logging.INFO)
transformers_logger = logging.getLogger("transformers")
transformers_logger.setLevel(logging.WARNING)

model = LanguageGenerationModel("gpt2", "gpt2")
model.generate("Let's give a minimal start to the model like")

Updated: