Construire un pipeline de génération de texte
Les pipelines Hugging Face facilitent l'utilisation des modèles d'apprentissage automatique pour diverses tâches. Dans cet exercice, vous allez créer un pipeline de génération de texte à l'aide du modèle gpt2 et personnaliser la sortie en ajustant ses paramètres.
N'hésitez pas à tester différentes suggestions dans le pipeline, telles que * « Et si… ? »,* * « Comment faire pour… ? »* ou toute autre idée créative que vous aimeriez explorer.
Cet exercice fait partie du cours
Travailler avec Hugging Face
Instructions
- Complétez le code manquant pour créer un pipeline de génération de texte à l'aide du modèle
"gpt2". - Veuillez fournir une phrase personnalisée de votre choix comme invite de saisie ; veuillez la garder courte afin d'éviter les délais d'expiration.
- Configurez le pipeline pour générer jusqu'à 10 jetons et produire 2 sorties.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
from transformers import ____
gpt2_pipeline = ____(task="____", model="openai-community/gpt2")
# Generate three text outputs with a maximum length of 10 tokens
results = gpt2_pipeline("What if AI", max_new_tokens=____, num_return_sequences=____)
for result in results:
print(result['generated_text'])