Doplňování textu s předtrénovanými modely GPT-2
Zpátky v PyBooks – tvůj aktuální projekt spočívá ve vytváření poutavých příběhů vycházejících z existujících textů, aby se zákazníkům nabídl zajímavější zážitek. K tomu potřebuješ výkonný nástroj pro generování textu, který dokáže plynule navazovat na zadaný začátek. Použiješ k tomu předtrénovaný model.
Následující knihovny a třídy jsou už načtené: torch a GPT2Tokenizer, GPT2LMHeadModel z balíčku transformers.
Toto cvičení je součástí kurzu
Deep Learning for Text with PyTorch
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Initialize the tokenizer
tokenizer = ____
# Initialize the pre-trained model
model = ____