model.tie_weights() should be applied after accelerator.prepare()

[Gladiator07]

What does this PR do?

Weight tying should be done after the model has been moved to XLA device as mentioned on PyTorch/XLA Troubleshooting guide here. This PR fixes this in examples/pytorch/language-modeling/run_mlm_no_trainer.py example script.

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you read the contributor guideline,
  Pull Request section?
• Was this discussed/approved via a Github issue or the forum? Please add a link
  to it if that's the case.
• Did you make sure to update the documentation with your changes? Here are the
  documentation guidelines, and
  here are tips on formatting docstrings.
• Did you write any new necessary tests?

Who can review ?

@sgugger @muellerzr

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[muellerzr]

Thanks! Indeed, this is the right way to go about this. In Accelerate it'd probably also be good to do this by default, similar to how Trainer does it

Could you run make style; make quality to pass the quality checks?

[muellerzr]

Thanks @Gladiator07 great work!