ERRORS: run_mlm_performer.py
## 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: tfs4.5.1
- Platform: ubuntu18.04
- Python version: 3.8
- PyTorch version (GPU?): 1.7
- Tensorflow version (GPU?): 2.3.0
- Using GPU in script?: yes
- Using distributed or parallel set-up in script?: no

I try to use the run_mlm_performer.py

`TOKENIZERS_PARALLELISM=false python run_mlm_performer.py --model_name_or_path ../cache_model/bert-base-chinese/ --tokenizer_name ../cache_model/bert-base-chinese/ --train_file ../data/wikicorpus_zh_one_article_per_line-jianti.txt --do_train --fp16 --output_dir ./test-mlm --max_seq_length 512 --per_device_train_batch_size 256 --reinitialize --overwrite_output_dir True --preprocessing_num_workers 8`

And
 raise AttributeError(
jax._src.traceback_util.FilteredStackTrace: AttributeError: "FlaxBertSelfAttention" object has no attribute "dropout_rate"

You will find that FlaxBertSelfAttention no defined dropout_rate...
I tried to use self.config.attention_probs_dropout_prob instead of config.dropout_rate. It seems worked.<|||||>This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/master/CONTRIBUTING.md) are likely to be ignored.