diff --git a/utils/notification_service.py b/utils/notification_service.py index 5c54809b262..43f132bf12d 100644 --- a/utils/notification_service.py +++ b/utils/notification_service.py @@ -1364,8 +1364,8 @@ if __name__ == "__main__": nvidia_daily_ci_workflow = "huggingface/transformers/.github/workflows/self-scheduled-caller.yml" amd_daily_ci_workflows = ( - "huggingface/transformers/.github/workflows/self-scheduled-amd-mi210-caller.yml", "huggingface/transformers/.github/workflows/self-scheduled-amd-mi250-caller.yml", + "huggingface/transformers/.github/workflows/self-scheduled-amd-mi300-caller.yml", ) is_nvidia_daily_ci_workflow = os.environ.get("GITHUB_WORKFLOW_REF").startswith(nvidia_daily_ci_workflow) is_amd_daily_ci_workflow = os.environ.get("GITHUB_WORKFLOW_REF").startswith(amd_daily_ci_workflows)