Skip to content

Commit 37ba57d

Browse files
committed
change workspace management
1 parent 8268039 commit 37ba57d

File tree

1 file changed

+6
-7
lines changed
  • terraform/modules/aws-sagemaker-hf-pytorch-inference-model-deployment

1 file changed

+6
-7
lines changed

terraform/modules/aws-sagemaker-hf-pytorch-inference-model-deployment/main.tf

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,14 @@
11
data "aws_caller_identity" "current" {}
22

3-
43
locals {
5-
env = terraform.workspace
64
account_id = data.aws_caller_identity.current.account_id
75
ecr_image_name = "763104351884.dkr.ecr.${var.aws_region}.amazonaws.com/huggingface-pytorch-inference"
86
ecr_image_tag = "${var.pytorch_version}-transformers${var.transformers_version}-cpu-${var.python_version}-ubuntu${var.ubuntu_version}"
97
}
108

9+
1110
resource "aws_iam_role" "model_execution_role" {
12-
name = "${local.env}-sagemaker-${var.model_name}-execution-role"
11+
name = "sagemaker-${var.model_name}-execution-role"
1312

1413
assume_role_policy = jsonencode({
1514
Version = "2012-10-17"
@@ -26,7 +25,7 @@ resource "aws_iam_role" "model_execution_role" {
2625
}
2726

2827
resource "aws_iam_policy" "s3_policy" {
29-
name = "${local.env}-sagemaker-${var.model_name}-s3-policy"
28+
name = "sagemaker-${var.model_name}-s3-policy"
3029

3130
policy = jsonencode({
3231
"Version" : "2012-10-17",
@@ -44,7 +43,7 @@ resource "aws_iam_policy" "s3_policy" {
4443
}
4544

4645
resource "aws_iam_policy" "ecr_policy" {
47-
name = "${local.env}-sagemaker-${var.model_name}-ecr-policy"
46+
name = "sagemaker-${var.model_name}-ecr-policy"
4847

4948
policy = jsonencode({
5049
"Version" : "2012-10-17",
@@ -92,7 +91,7 @@ resource "aws_sagemaker_model" "model" {
9291
}
9392

9493
resource "aws_sagemaker_endpoint_configuration" "model_endpoint_configuration" {
95-
name = "${local.env}-sagemaker-${var.model_name}-endpoint-config"
94+
name = "sagemaker-${var.model_name}-endpoint-config"
9695
production_variants {
9796
initial_variant_weight = 1
9897
model_name = aws_sagemaker_model.model.name
@@ -107,6 +106,6 @@ resource "aws_sagemaker_endpoint_configuration" "model_endpoint_configuration" {
107106

108107
resource "aws_sagemaker_endpoint" "model_endpoint" {
109108
endpoint_config_name = aws_sagemaker_endpoint_configuration.model_endpoint_configuration.name
110-
name = "${local.env}-sagemaker-${var.model_name}-endpoint"
109+
name = "sagemaker-${var.model_name}-endpoint"
111110
}
112111

0 commit comments

Comments
 (0)