transformers

242ec31a - [Flax] Refactor MLM (#12013)

[Flax] Refactor MLM (#12013) * fix_torch_device_generate_test * remove @ * finish refactor Co-authored-by: Patrick von Platen <patrick@huggingface.co>