CommencezCommencez gratuitement

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

Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
Modifier et exécuter le code