Update CI Docker base image for AMD tests

[ahadnagy]

What does this PR do?

This PR updates AMD CI image to use the latest PyTorch base image. This fix is required for the MI300 tests as PyTorch falls back to CPU on older versions.

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?

Transformers CI folks.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[ydshieh]

cc @ivarflakstad

[ydshieh]

OK, probably ivar is already on their time off.

We may want to check if this will work for other MIXXX.

[ydshieh]

I will merge (but please also take an eye to check if the image is built successfully tomorrow)