Hoàn thành văn bản với mô hình GPT-2 đã huấn luyện sẵn
Quay lại với PyBooks, dự án hiện tại của bạn là tạo ra những câu chuyện hấp dẫn dựa trên các truyện có sẵn để thu hút khách hàng và nâng cao trải nghiệm của họ. Để làm được điều này, bạn cần một công cụ sinh văn bản mạnh mẽ có thể tạo phần tiếp nối mạch lạc và cuốn hút. Bạn sẽ sử dụng một mô hình đã huấn luyện sẵn để thực hiện nhiệm vụ này.
Đã được nạp sẵn cho bạn: torch và GPT2Tokenizer,GPT2LMHeadModel từ transformers.
Bài tập này là một phần của khóa học
Deep Learning cho Văn bản với PyTorch
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Initialize the tokenizer
tokenizer = ____
# Initialize the pre-trained model
model = ____