From 323d6adb0e1fc36ec4da30832a36fcb91b4318f7 Mon Sep 17 00:00:00 2001 From: philschmid Date: Wed, 1 Dec 2021 10:36:27 +0100 Subject: [PATCH 1/2] added hugging face images to sagemaker pre-built ecr image --- .../prebuilt_ecr_image_data_source.go | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/internal/service/sagemaker/prebuilt_ecr_image_data_source.go b/internal/service/sagemaker/prebuilt_ecr_image_data_source.go index e2893cccf666..a37fcc51106e 100644 --- a/internal/service/sagemaker/prebuilt_ecr_image_data_source.go +++ b/internal/service/sagemaker/prebuilt_ecr_image_data_source.go @@ -70,6 +70,14 @@ const ( sageMakerRepositoryTensorFlowInferenceEIA = "tensorflow-inference-eia" // SageMaker Repo TensorFlow Training sageMakerRepositoryTensorFlowTraining = "tensorflow-training" + // SageMaker Repo HuggingFace TensorFlow Training + sageMakerRepositoryHuggingFaceTensorFlowTraining = "huggingface-tensorflow-training" + // SageMaker Repo HuggingFace TensorFlow Inference + sageMakerRepositoryHuggingFaceTensorFlowInference = "huggingface-tensorflow-inference" + // SageMaker Repo HuggingFace PyTorch Training + sageMakerRepositoryHuggingFacePyTorchTraining = "huggingface-pytorch-training" + // SageMaker Repo HuggingFace PyTorch Inference + sageMakerRepositoryHuggingFacePyTorchInference = "huggingface-pytorch-inference" ) // https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html @@ -295,6 +303,10 @@ func DataSourcePrebuiltECRImage() *schema.Resource { sageMakerRepositoryTensorFlowServing, sageMakerRepositoryTensorFlowServingEIA, sageMakerRepositoryTensorFlowTraining, + sageMakerRepositoryHuggingFaceTensorFlowTraining, + sageMakerRepositoryHuggingFaceTensorFlowInference, + sageMakerRepositoryHuggingFacePyTorchTraining, + sageMakerRepositoryHuggingFacePyTorchInference, sageMakerRepositoryXGBoost, }, false), }, @@ -367,7 +379,11 @@ func dataSourcePrebuiltECRImageRead(d *schema.ResourceData, meta interface{}) er sageMakerRepositoryPyTorchTraining, sageMakerRepositoryTensorFlowInference, sageMakerRepositoryTensorFlowInferenceEIA, - sageMakerRepositoryTensorFlowTraining: + sageMakerRepositoryTensorFlowTraining, + sageMakerRepositoryHuggingFaceTensorFlowTraining, + sageMakerRepositoryHuggingFaceTensorFlowInference, + sageMakerRepositoryHuggingFacePyTorchTraining, + sageMakerRepositoryHuggingFacePyTorchInference: id = sageMakerPrebuiltECRImageIDByRegion_DeepLearning[region] default: id = PrebuiltECRImageIDByRegion_FactorMachines[region] From f2453805737d9f0ec7dee901b51649a3b6e006a3 Mon Sep 17 00:00:00 2001 From: philschmid Date: Wed, 1 Dec 2021 10:51:08 +0100 Subject: [PATCH 2/2] add changelog --- .changelog/21983.txt | 3 +++ 1 file changed, 3 insertions(+) create mode 100644 .changelog/21983.txt diff --git a/.changelog/21983.txt b/.changelog/21983.txt new file mode 100644 index 000000000000..b76d5ad00c87 --- /dev/null +++ b/.changelog/21983.txt @@ -0,0 +1,3 @@ +```release-note:enhancement +data/aws_sagemaker_prebuilt_ecr_image: Add Hugging Face DLCs +``` \ No newline at end of file