diff --git a/src/sagemaker/image_uri_config/djl-deepspeed.json b/src/sagemaker/image_uri_config/djl-deepspeed.json index 133a744f6c..e98e382b0b 100644 --- a/src/sagemaker/image_uri_config/djl-deepspeed.json +++ b/src/sagemaker/image_uri_config/djl-deepspeed.json @@ -28,6 +28,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -60,6 +62,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -92,6 +96,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -124,6 +130,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -156,6 +164,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -188,6 +198,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -220,6 +232,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -252,6 +266,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -284,6 +300,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" diff --git a/src/sagemaker/image_uri_config/djl-fastertransformer.json b/src/sagemaker/image_uri_config/djl-fastertransformer.json index fd9ced32fe..b272460711 100644 --- a/src/sagemaker/image_uri_config/djl-fastertransformer.json +++ b/src/sagemaker/image_uri_config/djl-fastertransformer.json @@ -28,6 +28,10 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", + "us-iso-east-1": "886529160074", + "us-isob-east-1": "094389454867", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -60,6 +64,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -92,6 +98,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -124,6 +132,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" diff --git a/src/sagemaker/image_uri_config/djl-lmi.json b/src/sagemaker/image_uri_config/djl-lmi.json index 05632f9228..325327ca8e 100644 --- a/src/sagemaker/image_uri_config/djl-lmi.json +++ b/src/sagemaker/image_uri_config/djl-lmi.json @@ -28,6 +28,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -60,6 +62,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" diff --git a/src/sagemaker/image_uri_config/djl-neuronx.json b/src/sagemaker/image_uri_config/djl-neuronx.json index e01cb606e8..b5620c08fd 100644 --- a/src/sagemaker/image_uri_config/djl-neuronx.json +++ b/src/sagemaker/image_uri_config/djl-neuronx.json @@ -15,6 +15,10 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", + "us-iso-east-1": "886529160074", + "us-isob-east-1": "094389454867", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, @@ -33,6 +37,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, @@ -51,6 +57,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, @@ -87,6 +95,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, @@ -105,6 +115,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, @@ -123,6 +135,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, @@ -141,6 +155,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-2": "763104351884", "ca-west-1": "204538143572" }, diff --git a/src/sagemaker/image_uri_config/djl-tensorrtllm.json b/src/sagemaker/image_uri_config/djl-tensorrtllm.json index 835f9818e8..68db982632 100644 --- a/src/sagemaker/image_uri_config/djl-tensorrtllm.json +++ b/src/sagemaker/image_uri_config/djl-tensorrtllm.json @@ -28,6 +28,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -60,6 +62,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -92,6 +96,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -124,6 +130,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" @@ -156,6 +164,8 @@ "sa-east-1": "763104351884", "us-east-1": "763104351884", "us-east-2": "763104351884", + "us-gov-east-1": "446045086412", + "us-gov-west-1": "442386744353", "us-west-1": "763104351884", "us-west-2": "763104351884", "ca-west-1": "204538143572" diff --git a/src/sagemaker/jumpstart/constants.py b/src/sagemaker/jumpstart/constants.py index 47f4dd9b46..03a0ebe545 100644 --- a/src/sagemaker/jumpstart/constants.py +++ b/src/sagemaker/jumpstart/constants.py @@ -176,6 +176,12 @@ JumpStartLaunchedRegionInfo( region_name="cn-north-1", content_bucket="jumpstart-cache-prod-cn-north-1", + gated_content_bucket="jumpstart-private-cache-prod-cn-north-1", + ), + JumpStartLaunchedRegionInfo( + region_name="cn-northwest-1", + content_bucket="jumpstart-cache-prod-cn-northwest-1", + gated_content_bucket="jumpstart-private-cache-prod-cn-northwest-1", ), JumpStartLaunchedRegionInfo( region_name="il-central-1",