Vytvoření pipeline pro generování textu
Hugging Face pipelines usnadňují práci s modely strojového učení pro celou řadu úloh. V tomto cvičení sestavíš pipeline pro generování textu pomocí modelu gpt2 a přizpůsobíš výstup úpravou jeho parametrů.
Klidně experimentuj s různými vstupy, například "What if …?", "How to …?" nebo s jakýmkoli jiným nápadem, který tě napadne.
Toto cvičení je součástí kurzu
Working with Hugging Face
Pokyny k cvičení
- Doplň chybějící kód pro sestavení pipeline pro generování textu s použitím modelu
"gpt2". - Jako vstupní prompt zadej libovolnou vlastní větu; zvol kratší text, aby nedošlo k vypršení časového limitu.
- Nakonfiguruj pipeline tak, aby generovala maximálně 10 tokenů a produkovala 2 výstupy.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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'])