transformers

bc6ae0d5 - Update AMD docker image (rocm 6.1) (#35259)

Update AMD docker image (rocm 6.1) (#35259) * Use rocm 6.3 as base amd image and add nvidia-ml-py to exclude list * Align rocm base image with torch wheels @6.1. Seems like the most stable combo