diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 389bd905ccf0..2ddfe8ccb932 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -187,4 +187,4 @@ dlc-pr-tensorflow-2-eia-inference = "" dlc-pr-vllm = "" # sglang -dlc-pr-sglang = "" +dlc-pr-sglang = "" \ No newline at end of file diff --git a/test/test_utils/__init__.py b/test/test_utils/__init__.py index 53421e25ffe4..ce097ad9430a 100644 --- a/test/test_utils/__init__.py +++ b/test/test_utils/__init__.py @@ -142,7 +142,7 @@ def get_ami_id_ssm(region_name, parameter_path): ) AML2_CPU_ARM64_US_EAST_1 = get_ami_id_boto3( region_name="us-east-1", - ami_name_pattern="Deep Learning Base AMI (Amazon Linux 2) Version ??.?", + ami_name_pattern="Deep Learning ARM64 Base OSS Nvidia Driver GPU AMI (Amazon Linux 2) ????????", ) PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_EAST_1 = "ami-0673bb31cc62485dd" PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_WEST_2 = "ami-02d9a47bc61a31d43"