Complétion de texte avec des modèles GPT-2 préentraînés
De retour chez PyBooks, votre projet actuel consiste à créer des récits captivants à partir d'histoires existantes pour engager la clientèle et améliorer son expérience. Pour y parvenir, il vous faut un outil de génération de texte performant capable de produire, sans accroc, des suites de texte convaincantes. Vous utiliserez un modèle préentraîné pour accomplir la tâche.
Les éléments suivants ont été chargés pour vous : torch et GPT2Tokenizer,GPT2LMHeadModel de transformers.
Cette activité fait partie du cours
Apprentissage profond pour le texte avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Initialize the tokenizer
tokenizer = ____
# Initialize the pre-trained model
model = ____