75481


[Huggingface Transformers 入門 (6) - テキスト生成
](https://note.com/npaka/n/n5d296d8ae26d)

こちらの記事を真似てコードを動かしていますが、最初の

[code]

    import tensorflow as tf
    from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
    
[/code]

で以下のエラーが出ます

[code]

    ---------------------------------------------------------------------------
    ImportError                               Traceback (most recent call last)
    <ipython-input-23-4d671ac52c7c> in <module>()
          1 import tensorflow as tf
    ----> 2 from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
          3 
          4 
          5 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
    
    ImportError: cannot import name 'TFGPT2LMHeadModel' from 'transformers' (unknown location)
    
[/code]

色々調べて、tensoeflow==2.0.1を入れてみたのですが、動きませんでした。

ライブラリが古くて更新されているのでしょうか？  
環境はGoogle ColabのGPUを利用してtensorflowのバージョンを変更した以外はすべて同じとなっています。

詳しい方原因を教えていただきたいです。


CC BY-SA 4.0
2021-04-22T03:04:48.127
75443
2021-04-23T14:18:40.350
2021-04-22T03:51:37.583
3060
44467
question
python 機械学習 自然言語処理 google-colaboratory
transformersからTFGPT2LMHeadModelをインポートできない (cannot import name 'TFGPT2LMHeadModel' from 'transformers')


同様の報告がバグとして報告され、installationの記事が追加されたようです。  
<https://github.com/huggingface/transformers/issues/3396>

> This repository is tested on Python 3.6+, PyTorch 1.0.0+ (PyTorch 1.3.1+ for
> examples) and TensorFlow 2.0.

> You should install Transformers in a virtual environment. If you're
> unfamiliar with Python virtual environments, check out the user guide.

とあり、ほかの影響を受けない仮想環境に導入するように記載があります。  
実際いろいろがさがさと入っている私の環境では同様のエラーとなり、py venvで作った環境では問題なくimportできました。 情報共有まで。。


CC BY-SA 4.0
2021-04-23T14:18:40.350
75481
2021-04-23T14:18:40.350
10174
75443
answer