diff --git a/src/sagemaker/image_uri_config/djl-neuronx.json b/src/sagemaker/image_uri_config/djl-neuronx.json index 7b1fb0abba..8a59426ca9 100644 --- a/src/sagemaker/image_uri_config/djl-neuronx.json +++ b/src/sagemaker/image_uri_config/djl-neuronx.json @@ -9,6 +9,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -29,6 +31,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -49,6 +53,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -69,6 +75,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -89,6 +97,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -109,6 +119,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -129,6 +141,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", @@ -149,6 +163,8 @@ "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-west-1": "763104351884", "eu-west-3": "763104351884", diff --git a/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json b/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json index 3283dd21f0..c86a8dcf98 100644 --- a/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json +++ b/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json @@ -18,6 +18,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -27,6 +29,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" }, @@ -47,6 +51,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -56,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-2": "763104351884", "ca-west-1": "204538143572" }, @@ -76,6 +84,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -85,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 +117,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -114,6 +128,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" }, @@ -134,6 +150,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -143,6 +161,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" }, @@ -163,6 +183,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -172,6 +194,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" }, @@ -192,6 +216,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -221,6 +247,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -230,6 +258,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" }, @@ -250,6 +280,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -259,6 +291,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/huggingface-neuron.json b/src/sagemaker/image_uri_config/huggingface-neuron.json index ae38ce209b..623ecf70e3 100644 --- a/src/sagemaker/image_uri_config/huggingface-neuron.json +++ b/src/sagemaker/image_uri_config/huggingface-neuron.json @@ -23,6 +23,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -32,6 +34,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/huggingface-neuronx.json b/src/sagemaker/image_uri_config/huggingface-neuronx.json index 3721d75c5f..5b45e37586 100644 --- a/src/sagemaker/image_uri_config/huggingface-neuronx.json +++ b/src/sagemaker/image_uri_config/huggingface-neuronx.json @@ -25,6 +25,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -34,6 +36,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" }, @@ -61,6 +65,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -70,6 +76,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" }, @@ -97,6 +105,8 @@ "ap-southeast-1": "763104351884", "ap-southeast-2": "763104351884", "ap-southeast-4": "457447274322", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-south-2": "503227376785", @@ -106,6 +116,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/huggingface-training-compiler.json b/src/sagemaker/image_uri_config/huggingface-training-compiler.json index 735e7917b3..59a3c21143 100644 --- a/src/sagemaker/image_uri_config/huggingface-training-compiler.json +++ b/src/sagemaker/image_uri_config/huggingface-training-compiler.json @@ -70,6 +70,8 @@ "ap-southeast-3": "907027046896", "ap-southeast-4": "457447274322", "ca-central-1": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-north-1": "763104351884", @@ -83,6 +85,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" @@ -110,6 +114,8 @@ "ap-southeast-3": "907027046896", "ap-southeast-4": "457447274322", "ca-central-1": "763104351884", + "cn-north-1": "727897471807", + "cn-northwest-1": "727897471807", "eu-central-1": "763104351884", "eu-central-2": "380420809688", "eu-north-1": "763104351884", @@ -123,6 +129,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" @@ -168,6 +176,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",