transformers
09a71e5b - Move the MI300 docker image to ROCm 10.1 and update test expectations (#49460)

Commit
3 days ago
Move the MI300 docker image to ROCm 10.1 and update test expectations (#49460) * [AMD CI] Move the AMD docker image to ROCm 10 / torch 2.12 / Python 3.14 Base: rocm/vllm:rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.0 (ships flash-attn). - drop the separate flash-attn wheel and the system python apt packages - torchcodec 0.17.0 CPU build (the PyPI wheel needs CUDA libnvrtc) - build kenlm from source with a regenerated Cython file (no py3.14 build on PyPI) - import check at the end of the build * Update ROCm (MI300) test expectations for the ROCm 10 image * Remove Dockerfile comment * [AMD CI] Move the AMD docker image to ROCm 10.1 / torch 2.13 and pin MIOpen conv selection * Install on top of the ROCm 10.1 image's Debian Python through a venv * Update ROCm (MI300) test expectations for the ROCm 10.1 image * Shorten Dockerfile comments
Parents
Loading