diff --git a/terraform/2_sagemaker/main.tf b/terraform/2_sagemaker/main.tf index c7bd935e..5bb25330 100644 --- a/terraform/2_sagemaker/main.tf +++ b/terraform/2_sagemaker/main.tf @@ -48,7 +48,7 @@ resource "aws_sagemaker_model" "embedding_model" { execution_role_arn = aws_iam_role.sagemaker_role.arn primary_container { - image = var.sagemaker_image_uri + image = "763104351884.dkr.ecr.${var.aws_region}.amazonaws.com/huggingface-pytorch-inference:1.13.1-transformers4.26.0-cpu-py39-ubuntu20.04" environment = { HF_MODEL_ID = var.embedding_model_name HF_TASK = "feature-extraction" diff --git a/terraform/2_sagemaker/variables.tf b/terraform/2_sagemaker/variables.tf index 313102e9..17abb24a 100644 --- a/terraform/2_sagemaker/variables.tf +++ b/terraform/2_sagemaker/variables.tf @@ -6,7 +6,6 @@ variable "aws_region" { variable "sagemaker_image_uri" { description = "URI of the SageMaker container image" type = string - default = "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-inference:1.13.1-transformers4.26.0-cpu-py39-ubuntu20.04" } variable "embedding_model_name" {