diff --git a/.github/workflows/self-scheduled-amd-mi355-caller.yml b/.github/workflows/self-scheduled-amd-mi355-caller.yml index 19371aacf521..bd2fde3b0529 100644 --- a/.github/workflows/self-scheduled-amd-mi355-caller.yml +++ b/.github/workflows/self-scheduled-amd-mi355-caller.yml @@ -21,7 +21,7 @@ jobs: job: run_models_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-amd-gpu + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -33,7 +33,7 @@ jobs: job: run_pipelines_torch_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-amd-gpu + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -45,7 +45,7 @@ jobs: job: run_examples_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-amd-gpu + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -57,7 +57,7 @@ jobs: job: run_torch_cuda_extensions_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-deepspeed-amd-gpu + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit