Mulai sekarangMulai gratis

Text completion with pre-trained GPT-2 models

Back at PyBooks, your current project involves creating captivating narratives based on existing stories to engage customers and enhance their experience. To achieve this, you need a powerful text generation tool that can seamlessly generate compelling text continuations. You'll be using a pre-trained model to get the job done.

The following has been loaded for you: torch and GPT2Tokenizer,GPT2LMHeadModel from transformers.

Latihan ini merupakan bagian dari kursus

Deep Learning for Text with PyTorch

Lihat Kursus

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
Edit dan Jalankan Kode