1
1
data "aws_caller_identity" "current" {}
2
2
3
-
4
3
locals {
5
- env = terraform. workspace
6
4
account_id = data. aws_caller_identity . current . account_id
7
5
ecr_image_name = " 763104351884.dkr.ecr.${ var . aws_region } .amazonaws.com/huggingface-pytorch-inference"
8
6
ecr_image_tag = " ${ var . pytorch_version } -transformers${ var . transformers_version } -cpu-${ var . python_version } -ubuntu${ var . ubuntu_version } "
9
7
}
10
8
9
+
11
10
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"
13
12
14
13
assume_role_policy = jsonencode ({
15
14
Version = " 2012-10-17"
@@ -26,7 +25,7 @@ resource "aws_iam_role" "model_execution_role" {
26
25
}
27
26
28
27
resource "aws_iam_policy" "s3_policy" {
29
- name = " ${ local . env } - sagemaker-${ var . model_name } -s3-policy"
28
+ name = " sagemaker-${ var . model_name } -s3-policy"
30
29
31
30
policy = jsonencode ({
32
31
" Version" : " 2012-10-17" ,
@@ -44,7 +43,7 @@ resource "aws_iam_policy" "s3_policy" {
44
43
}
45
44
46
45
resource "aws_iam_policy" "ecr_policy" {
47
- name = " ${ local . env } - sagemaker-${ var . model_name } -ecr-policy"
46
+ name = " sagemaker-${ var . model_name } -ecr-policy"
48
47
49
48
policy = jsonencode ({
50
49
" Version" : " 2012-10-17" ,
@@ -92,7 +91,7 @@ resource "aws_sagemaker_model" "model" {
92
91
}
93
92
94
93
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"
96
95
production_variants {
97
96
initial_variant_weight = 1
98
97
model_name = aws_sagemaker_model. model . name
@@ -107,6 +106,6 @@ resource "aws_sagemaker_endpoint_configuration" "model_endpoint_configuration" {
107
106
108
107
resource "aws_sagemaker_endpoint" "model_endpoint" {
109
108
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"
111
110
}
112
111
0 commit comments