transformersからTFGPT2LMHeadModelをインポートできない(cannot import name 'TFGPT2LMHeadModel' from 'transformers')件について - Qiita

はじめに 最近技術調査でHuggingFaceのサンプルコードを動かしているのですが、なかなか動かないことが多く(公式のものでも)、今回は動くようにするところまでできたので、まとめたいと思います。 問題 こちらの記事を参考にハンズオンをGoogle Colabratory`で行っていたところ、 import tensorflow as tf from transformers import TFGPT2LMHeadModel, GPT2Tokenizer tokenizer = GPT2Tokenizer.from_pretrained("gpt2") # 警告を避けるために、EOS トークンを PAD トークンとして追加 model = TFGPT2LMHeadModel.from_pretrained("gpt2", pad_token_id=tokenizer.eos_token_i