ImportError: cannot import name 'TFLongformerForMaskedLM'

I am trying to follow the example mentioned in https://huggingface.co/transformers/model_doc/longformer.html#tflongformerformaskedlm.

import tensorflow as tf
from transformers import LongformerTokenizer, TFLongformerForMaskedLM

I get below error:


ImportError Traceback (most recent call last)
in
1 import tensorflow as tf
----> 2 from transformers import LongformerTokenizer, TFLongformerForMaskedLM

ImportError: cannot import name ‘TFLongformerForMaskedLM’

How to resolve this? Pls let me know.

Thanks,

Maybe you should check if you have the latest version of Huggingface Transformers. (i.e. 3.4.0)
I can import it without any problem :slight_smile:

Thanks for replying.

!pip freeze | grep transformers
transformers==3.4.0

And, I still get the same error

import tensorflow as tf
from transformers import LongformerTokenizer, TFLongformerForMaskedLM


ImportError Traceback (most recent call last)
in
1 import tensorflow as tf
----> 2 from transformers import LongformerTokenizer, TFLongformerForMaskedLM

ImportError: cannot import name ‘TFLongformerForMaskedLM’

I have no idea why you did get this import error.
Maybe you can re-check with my dummy Colab in the 2nd cell with showing successfully creating the model.

(The 3rd to last cells are not related to TFLongformer, please ignore)