Skip to content

Commit af5e42d

Browse files
authored
fix:fix local deploy bug (#119)
* 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 * add glm4-z1 models * modify vllm backend * add qwen3 * fix cli bugs * fix * add deeseek r1/Qwen3-235B-A22B * fix local deploy account bug
1 parent 1af989c commit af5e42d

File tree

2 files changed

+13
-9
lines changed

2 files changed

+13
-9
lines changed

src/emd/models/model.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -375,13 +375,13 @@ def get_deploy_version_from_stack_name(cls,stack_name):
375375
except Exception as e:
376376
raise ValueError(f"stack_name:{stack_name} is not a valid model stack name")
377377

378-
def get_image_build_account_id(self):
379-
current_account_id = boto3.client("sts").get_caller_identity()["Account"]
380-
build_image_account_id = (
381-
self.executable_config.current_engine.base_image_account_id or \
382-
current_account_id
383-
)
384-
return build_image_account_id
378+
# def get_image_build_account_id(self):
379+
# current_account_id = boto3.client("sts").get_caller_identity()["Account"]
380+
# build_image_account_id = (
381+
# self.executable_config.current_engine.base_image_account_id or \
382+
# current_account_id
383+
# )
384+
# return build_image_account_id
385385

386386
def get_image_push_account_id(self):
387387
current_account_id = boto3.client("sts").get_caller_identity()["Account"]

src/pipeline/deploy/build_and_push_image.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,11 @@ def run(
194194

195195
# docker build image
196196
# get current aws account_id
197-
push_image_account_id = execute_model.get_image_push_account_id()
197+
if instance_type == InstanceType.LOCAL:
198+
push_image_account_id = "local"
199+
else:
200+
push_image_account_id = execute_model.get_image_push_account_id()
201+
198202
build_image_account_id = (
199203
execute_model.executable_config.current_engine.base_image_account_id
200204
)
@@ -325,7 +329,7 @@ def run(
325329
logger.info(f"pushing image: {push_image_script}")
326330
assert os.system(push_image_script) == 0
327331

328-
image_uri = ecr_repo_uri
332+
# image_uri = ecr_repo_uri
329333
logger.info(f"Image URI: {ecr_repo_uri}")
330334

331335
parameters = {"ecr_repo_uri": ecr_repo_uri}

0 commit comments

Comments
 (0)