Membangun pipeline generasi teks
Pipeline Hugging Face memudahkan penggunaan model machine learning untuk berbagai tugas. Pada latihan ini, Anda akan membangun pipeline generasi teks menggunakan model gpt2 dan menyesuaikan keluarannya dengan mengatur parameternya.
Silakan bereksperimen dengan berbagai prompt dalam pipeline, seperti "Bagaimana jika …?", "Cara …?", atau ide kreatif lain yang ingin Anda jelajahi.
Latihan ini adalah bagian dari kursus
Bekerja dengan Hugging Face
Petunjuk latihan
- Lengkapi kode yang hilang untuk membangun pipeline generasi teks menggunakan model
"gpt2". - Berikan kalimat khusus pilihan Anda sebagai prompt masukan; buat singkat untuk mencegah batas waktu habis.
- Konfigurasikan pipeline untuk menghasilkan hingga 10 token dan menghasilkan 2 keluaran.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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'])