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")