開始使用免費開始

使用預訓練 GPT-2 模型進行文字補全

回到 PyBooks,你目前的專案是根據既有故事創作引人入勝的延伸敘事,以吸引顧客並提升體驗。為了達成這個目標,你需要一個強大的文字生成工具,能夠順暢地產生有說服力的續寫內容。你將使用預訓練模型來完成這項工作。

以下內容已為你載入:torch,以及 transformers 中的 GPT2TokenizerGPT2LMHeadModel

本練習屬於課程

Deep Learning for Text with PyTorch

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
編輯並執行程式碼