mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
52 lines
1.6 KiB
Bash
Executable File
52 lines
1.6 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
# If you want to build a specific flavor, set the CONTAINER_FLAVOR environment variable
|
|
# e.g. CONTAINER_FLAVOR=cpu ./build.sh
|
|
# Possible Values are:
|
|
# - cpu
|
|
# - cuda
|
|
# - rocm
|
|
# Don't forget to also set it when executing run.sh
|
|
# if it is not set, the script will try to detect the flavor by itself.
|
|
#
|
|
# Doc can be found here:
|
|
# https://invoke-ai.github.io/InvokeAI/installation/040_INSTALL_DOCKER/
|
|
|
|
SCRIPTDIR=$(dirname "${BASH_SOURCE[0]}")
|
|
cd "$SCRIPTDIR" || exit 1
|
|
|
|
source ./env.sh
|
|
|
|
DOCKERFILE=${INVOKE_DOCKERFILE:-./Dockerfile}
|
|
|
|
# print the settings
|
|
echo -e "You are using these values:\n"
|
|
echo -e "Dockerfile:\t\t${DOCKERFILE}"
|
|
echo -e "index-url:\t\t${PIP_EXTRA_INDEX_URL:-none}"
|
|
echo -e "Volumename:\t\t${VOLUMENAME}"
|
|
echo -e "Platform:\t\t${PLATFORM}"
|
|
echo -e "Container Registry:\t${CONTAINER_REGISTRY}"
|
|
echo -e "Container Repository:\t${CONTAINER_REPOSITORY}"
|
|
echo -e "Container Tag:\t\t${CONTAINER_TAG}"
|
|
echo -e "Container Flavor:\t${CONTAINER_FLAVOR}"
|
|
echo -e "Container Image:\t${CONTAINER_IMAGE}\n"
|
|
|
|
# Create docker volume
|
|
if [[ -n "$(docker volume ls -f name="${VOLUMENAME}" -q)" ]]; then
|
|
echo -e "Volume already exists\n"
|
|
else
|
|
echo -n "creating docker volume "
|
|
docker volume create "${VOLUMENAME}"
|
|
fi
|
|
|
|
# Build Container
|
|
docker build \
|
|
--platform="${PLATFORM:-linux/amd64}" \
|
|
--tag="${CONTAINER_IMAGE:-invokeai}" \
|
|
${CONTAINER_FLAVOR:+--build-arg="CONTAINER_FLAVOR=${CONTAINER_FLAVOR}"} \
|
|
${PIP_EXTRA_INDEX_URL:+--build-arg="PIP_EXTRA_INDEX_URL=${PIP_EXTRA_INDEX_URL}"} \
|
|
${PIP_PACKAGE:+--build-arg="PIP_PACKAGE=${PIP_PACKAGE}"} \
|
|
--file="${DOCKERFILE}" \
|
|
..
|