77 lines
1.9 KiB
Bash
Executable File
77 lines
1.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# This script shows how to build the Docker image and push it to ECR to be ready for use
|
|
# by Bracket.
|
|
|
|
# The argument to this script is the image name. This will be used as the image on the local
|
|
# machine and combined with the account and region to form the repository name for ECR.
|
|
image=$1
|
|
tag=$2
|
|
|
|
if [ "$image" = "" ]
|
|
then
|
|
echo "Usage: $0 <image-name>"
|
|
exit 1
|
|
fi
|
|
|
|
if [ "$tag" = "" ]
|
|
then
|
|
tag=latest
|
|
fi
|
|
|
|
# Get the account number associated with the current IAM credentials
|
|
account=$(aws sts get-caller-identity --query Account --output text)
|
|
|
|
if [ $? -ne 0 ]
|
|
then
|
|
exit 255
|
|
fi
|
|
|
|
|
|
# Get the region defined in the current configuration (default to us-west-2 if none defined)
|
|
region=$(aws configure get region)
|
|
# region=${region:-us-west-2}
|
|
|
|
|
|
|
|
# If the repository doesn't exist in ECR, create it.
|
|
image_name="stable-diffusion-aws-extension/${image}"
|
|
fullname="${account}.dkr.ecr.${region}.amazonaws.com/${image_name}:${tag}"
|
|
|
|
desc_output=$(aws ecr describe-repositories --repository-names ${image_name} 2>&1)
|
|
|
|
if [ $? -ne 0 ]
|
|
then
|
|
if echo ${desc_output} | grep -q RepositoryNotFoundException
|
|
then
|
|
aws ecr create-repository --repository-name "${image_name}" > /dev/null
|
|
else
|
|
>&2 echo ${desc_output}
|
|
fi
|
|
fi
|
|
|
|
aws ecr get-login-password --region ${region} | docker login -u AWS --password-stdin ${account}.dkr.ecr.${region}.amazonaws.com
|
|
|
|
# if [ "$image" == "aigc-webui-utils" ]; then
|
|
# repo_id="e2t2y5y0"
|
|
# elif [ "$image" == "aigc-webui-inference" ]; then
|
|
# repo_id="l7s6x2w8"
|
|
# elif [ "$image" == "aigc-webui-dreambooth-train" ]; then
|
|
# repo_id="e2t2y5y0"
|
|
# fi
|
|
|
|
repo_name=${image}
|
|
complete_command="FROM public.ecr.aws/aws-gcr-solutions/stable-diffusion-aws-extension/${repo_name}:${tag}"
|
|
|
|
echo $complete_command
|
|
|
|
echo $complete_command > Dockerfile
|
|
|
|
docker logout public.ecr.aws
|
|
|
|
docker build -t ${image_name} -f Dockerfile .
|
|
docker tag ${image_name} ${fullname}
|
|
|
|
docker push ${fullname}
|
|
echo $fullname
|