an unexpected keyword argument 'output_signature'
## Environment info
<!-- You can run the command `transformers-cli env` and copy-and-paste its output below.
     Don't forget to fill out the missing fields in that output! -->

- `transformers` version:4.9.0
- Platform:linux
- Python version:3.6
- PyTorch version (GPU?):1.7
- Tensorflow version (GPU?):2.3,gpu version
- Using GPU in script?:k40m
- Using distributed or parallel set-up in script?:no


when I execute
python run_mlm.py 
--model_name_or_path bert-base-chinese 
--output_dir /data/bert_virtual/modelinfo/ 
--train_file /data/zxdata/000002_0.txt
from examples/tensorflow/language-modeling,this error happens

Traceback (most recent call last):
File "run_mlm.py", line 619, in <module>
main()
File "run_mlm.py", line 543, in main
tf.data.Dataset.from_generator(train_generator, output_signature=train_signature)
TypeError: from_generator() got an unexpected keyword argument 'output_signature'



update tensorflow to 2.5 solve this.<|||||>Use tf >= v2.4.0 to solve this issue. This was in the release notes.