Kom igångKom igång gratis

Textkomplettering med förtränade GPT-2-modeller

Tillbaka på PyBooks handlar ditt nuvarande projekt om att skapa engagerande berättelser baserade på befintliga historier – för att fånga kundernas intresse och förbättra deras upplevelse. Du behöver ett kraftfullt textgenereringsverktyg som smidigt kan generera övertygande textfortsättningar. Du kommer att använda en förtränad modell för att lösa uppgiften.

Följande har laddats in åt dig: torch samt GPT2Tokenizer och GPT2LMHeadModel från transformers.

Den här övningen är en del av kursen

Djupinlärning för text med PyTorch

Visa kurs

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
Redigera och kör kod