mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
85 lines
2.7 KiB
Bash
Executable File
85 lines
2.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
# IMPORTANT: You need to have a token on huggingface.co to be able to download the checkpoint!!!
|
|
# configure values by using env when executing build.sh
|
|
# f.e. env ARCH=aarch64 GITHUB_INVOKE_AI=https://github.com/yourname/yourfork.git ./build.sh
|
|
|
|
source ./docker-build/env.sh || echo "please run from repository root" || exit 1
|
|
|
|
invokeai_conda_version=${INVOKEAI_CONDA_VERSION:-py39_4.12.0-${platform/\//-}}
|
|
invokeai_conda_prefix=${INVOKEAI_CONDA_PREFIX:-\/opt\/conda}
|
|
invokeai_conda_env_file=${INVOKEAI_CONDA_ENV_FILE:-environment-lin-cuda.yml}
|
|
invokeai_git=${INVOKEAI_GIT:-invoke-ai/InvokeAI}
|
|
invokeai_branch=${INVOKEAI_BRANCH:-main}
|
|
huggingface_token=${HUGGINGFACE_TOKEN?}
|
|
|
|
# print the settings
|
|
echo "You are using these values:"
|
|
echo -e "project_name:\t\t ${project_name}"
|
|
echo -e "volumename:\t\t ${volumename}"
|
|
echo -e "arch:\t\t\t ${arch}"
|
|
echo -e "platform:\t\t ${platform}"
|
|
echo -e "invokeai_conda_version:\t ${invokeai_conda_version}"
|
|
echo -e "invokeai_conda_prefix:\t ${invokeai_conda_prefix}"
|
|
echo -e "invokeai_conda_env_file: ${invokeai_conda_env_file}"
|
|
echo -e "invokeai_git:\t\t ${invokeai_git}"
|
|
echo -e "invokeai_tag:\t\t ${invokeai_tag}\n"
|
|
|
|
_runAlpine() {
|
|
docker run \
|
|
--rm \
|
|
--interactive \
|
|
--tty \
|
|
--mount source="$volumename",target=/data \
|
|
--workdir /data \
|
|
alpine "$@"
|
|
}
|
|
|
|
_copyCheckpoints() {
|
|
echo "creating subfolders for models and outputs"
|
|
_runAlpine mkdir models
|
|
_runAlpine mkdir outputs
|
|
echo "downloading v1-5-pruned-emaonly.ckpt"
|
|
_runAlpine wget \
|
|
--header="Authorization: Bearer ${huggingface_token}" \
|
|
-O models/v1-5-pruned-emaonly.ckpt \
|
|
https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt
|
|
echo "done"
|
|
}
|
|
|
|
_checkVolumeContent() {
|
|
_runAlpine ls -lhA /data/models
|
|
}
|
|
|
|
_getModelMd5s() {
|
|
_runAlpine \
|
|
alpine sh -c "md5sum /data/models/*.ckpt"
|
|
}
|
|
|
|
if [[ -n "$(docker volume ls -f name="${volumename}" -q)" ]]; then
|
|
echo "Volume already exists"
|
|
if [[ -z "$(_checkVolumeContent)" ]]; then
|
|
echo "looks empty, copying checkpoint"
|
|
_copyCheckpoints
|
|
fi
|
|
echo "Models in ${volumename}:"
|
|
_checkVolumeContent
|
|
else
|
|
echo -n "createing docker volume "
|
|
docker volume create "${volumename}"
|
|
_copyCheckpoints
|
|
fi
|
|
|
|
# Build Container
|
|
docker build \
|
|
--platform="${platform}" \
|
|
--tag "${invokeai_tag}" \
|
|
--build-arg project_name="${project_name}" \
|
|
--build-arg conda_version="${invokeai_conda_version}" \
|
|
--build-arg conda_prefix="${invokeai_conda_prefix}" \
|
|
--build-arg conda_env_file="${invokeai_conda_env_file}" \
|
|
--build-arg invokeai_git="${invokeai_git}" \
|
|
--build-arg invokeai_branch="${invokeai_branch}" \
|
|
--file ./docker-build/Dockerfile \
|
|
.
|