Zacznij terazZacznij za darmo

Uzupełnianie tekstu za pomocą wstępnie wytrenowanych modeli GPT-2

W PyBooks pracujesz nad projektem, którego celem jest tworzenie wciągających narracji na podstawie istniejących historii – tak, aby zaangażować klientów i wzbogacić ich doświadczenia. Do tego potrzebujesz wydajnego narzędzia do generowania tekstu, które płynnie stworzy przekonujące kontynuacje. Wykorzystasz do tego celu wstępnie wytrenowany model.

Następujące elementy zostały już załadowane: torch oraz GPT2Tokenizer i GPT2LMHeadModel z biblioteki transformers.

To ćwiczenie jest częścią kursu

Uczenie głębokie dla tekstu z PyTorch

Zobacz kurs

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
Edytuj i uruchom kod