Завершення тексту за допомогою попередньо натренованих моделей GPT-2
Повернімося до PyBooks: ваш поточний проєкт полягає у створенні захопливих оповідей на основі наявних історій, щоб зацікавити клієнтів і поліпшити їхній досвід. Для цього вам потрібен потужний інструмент генерування тексту, який зможе плавно створювати переконливі продовження. Ви використаєте попередньо натреновану модель, щоб виконати це завдання.
Для вас уже завантажено: torch і GPT2Tokenizer,GPT2LMHeadModel з transformers.
Ця вправа є частиною курсу
Глибоке навчання для тексту з PyTorch
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Initialize the tokenizer
tokenizer = ____
# Initialize the pre-trained model
model = ____