diff --git a/utils/notification_service.py b/utils/notification_service.py index b6d4e8b84c..9daceacb35 100644 --- a/utils/notification_service.py +++ b/utils/notification_service.py @@ -1397,8 +1397,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-mi300-caller.yml", "huggingface/transformers/.github/workflows/self-scheduled-amd-mi325-caller.yml", + "huggingface/transformers/.github/workflows/self-scheduled-amd-mi355-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)