Skip to content

Commit 257323f

Browse files
authored
feat:add some ecs params (#98)
* merge * merge * add Mistral-Small-3.1-24B-Instruct-2503 * modify qwq-32b deploy * add txgemma model; * modify model list command * fix typo * add some ecs parameters
1 parent abaf04b commit 257323f

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

src/emd/cfn/ecs/post_build.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,9 @@ def deploy_vpc_template(region):
104104
vpc_id = None
105105
subnets = None
106106
for output in outputs:
107-
if output["OutputKey"] == "VPCID":
107+
if output["OutputKey"] == "VPCID" and output["OutputValue"]:
108108
vpc_id = output["OutputValue"]
109-
elif output["OutputKey"] == "Subnets":
109+
elif output["OutputKey"] == "Subnets" and output["OutputValue"]:
110110
subnets = output["OutputValue"]
111111
update_parameters_file("parameters.json", {"VPCID": vpc_id, "Subnets": subnets})
112112
return vpc_id, subnets

src/emd/models/services.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,9 @@
9292
"EngineType":"engine_type",
9393
"Region": "region",
9494
"DesiredCapacity": ValueWithDefault(name="desired_capacity",default=1),
95+
"MaxSize": ValueWithDefault(name="max_size",default=1),
96+
"VPCID": ValueWithDefault(name="vpc_id",default=""),
97+
"Subnets": ValueWithDefault(name="subnet_ids",default=""),
9598
"ContainerCpu": "container_cpu",
9699
"ContainerMemory": "container_memory",
97100
"ContainerGpu":"instance_gpu_num"

src/pipeline/deploy/deploy.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,9 +108,7 @@ def run(
108108
+ "-"
109109
+ time.strftime("%Y-%m-%d-%H-%M-%S")
110110
)
111-
112111
role_arn = get_or_create_role(role_name, region)
113-
114112
create_sagemaker_endpoint(
115113
region=region,
116114
instance_type=instance_type,

0 commit comments

Comments
 (0)