From bf7c4f128c89bec0eb55269608c9f3eb2fb08b20 Mon Sep 17 00:00:00 2001 From: txlian Date: Wed, 4 Sep 2024 19:01:50 -0400 Subject: [PATCH] add us-gov region repo accounts to djl image metadata --- .../image_uri_config/djl-deepspeed.json | 18 ++++++++++++++++++ .../djl-fastertransformer.json | 8 ++++++++ src/sagemaker/image_uri_config/djl-lmi.json | 4 ++++ .../image_uri_config/djl-neuronx.json | 16 ++++++++++++++++ .../image_uri_config/djl-tensorrtllm.json | 10 ++++++++++ 5 files changed, 56 insertions(+) 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..f8924a3c46 100644 --- a/src/sagemaker/image_uri_config/djl-fastertransformer.json +++ b/src/sagemaker/image_uri_config/djl-fastertransformer.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" 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..7b1fb0abba 100644 --- a/src/sagemaker/image_uri_config/djl-neuronx.json +++ b/src/sagemaker/image_uri_config/djl-neuronx.json @@ -15,6 +15,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" }, @@ -33,6 +35,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 +55,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" }, @@ -69,6 +75,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"