使用預訓練 GPT-2 模型進行文字補全
回到 PyBooks,你目前的專案是根據既有故事創作引人入勝的延伸敘事,以吸引顧客並提升體驗。為了達成這個目標,你需要一個強大的文字生成工具,能夠順暢地產生有說服力的續寫內容。你將使用預訓練模型來完成這項工作。
以下內容已為你載入:torch,以及 transformers 中的 GPT2Tokenizer、GPT2LMHeadModel。
本練習屬於課程
Deep Learning for Text with PyTorch
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Initialize the tokenizer
tokenizer = ____
# Initialize the pre-trained model
model = ____