add json

2024-06-21 12:04:12 +08:00 · 2024-06-21 12:04:12 +08:00 · bda5deefc3
parent 8c091867a4
commit bda5deefc3
1 changed files with 2 additions and 1 deletions
--- a/docker_start.sh
+++ b/docker_start.sh
@ -82,7 +82,8 @@ WORKDIR /home/ubuntu/ComfyUI"
  START_HANDLER="#!/bin/bash
 set -euxo pipefail

-WORKFLOW_NAME=\$(cat $CONTAINER_PATH/$PROGRAM_NAME)
+export ACCOUNT_ID=$(aws sts get-caller-identity --query "Account" --output text)
+export WORKFLOW_NAME=\$(cat $CONTAINER_PATH/$PROGRAM_NAME)

 if [ \"\$WORKFLOW_NAME\" = \"default\" ]; then
  BASE_IMAGE=$PUBLIC_BASE_IMAGE