mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
66 lines
2.0 KiB
Bash
66 lines
2.0 KiB
Bash
|
#!/bin/bash
|
||
|
|
set -e -o pipefail
|
||
|
|
|
||
|
|
### Container entrypoint
|
||
|
|
# Runs the CMD as defined by the Dockerfile or passed to `docker run`
|
||
|
|
# Can be used to configure the runtime dir
|
||
|
|
# Bypass by using ENTRYPOINT or `--entrypoint`
|
||
|
|
|
||
|
|
### Set INVOKEAI_ROOT pointing to a valid runtime directory
|
||
|
|
# Otherwise configure the runtime dir first.
|
||
|
|
|
||
|
|
### Configure the InvokeAI runtime directory (done by default)):
|
||
|
|
# docker run --rm -it <this image> --configure
|
||
|
|
# or skip with --no-configure
|
||
|
|
|
||
|
|
### Set the CONTAINER_UID envvar to match your user.
|
||
|
|
# Ensures files created in the container are owned by you:
|
||
|
|
# docker run --rm -it -v /some/path:/invokeai -e CONTAINER_UID=$(id -u) <this image>
|
||
|
|
# Default UID: 1000 chosen due to popularity on Linux systems. Possibly 501 on MacOS.
|
||
|
|
|
||
|
|
USER_ID=${CONTAINER_UID:-1000}
|
||
|
|
USER=invoke
|
||
|
|
usermod -u ${USER_ID} ${USER} 1>/dev/null
|
||
|
|
|
||
|
|
configure() {
|
||
|
|
# Configure the runtime directory
|
||
|
|
if [[ -f ${INVOKEAI_ROOT}/invokeai.yaml ]]; then
|
||
|
|
echo "${INVOKEAI_ROOT}/invokeai.yaml found."
|
||
|
|
echo "To reconfigure InvokeAI, please delete it."
|
||
|
|
echo "==========================================="
|
||
|
|
else
|
||
|
|
mkdir -p ${INVOKEAI_ROOT}
|
||
|
|
chown --recursive ${USER} ${INVOKEAI_ROOT}
|
||
|
|
gosu ${USER} invokeai-configure --yes
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
## Skip attempting to configure.
|
||
|
|
## Must be passed first, before any other args.
|
||
|
|
if [[ $1 != "--no-configure" ]]; then
|
||
|
|
configure
|
||
|
|
else
|
||
|
|
shift
|
||
|
|
fi
|
||
|
|
|
||
|
|
### Set the $PUBLIC_KEY env var to enable SSH access.
|
||
|
|
# We do not install openssh-server in the image by default to avoid bloat.
|
||
|
|
# but it is useful to have the full SSH server e.g. on Runpod.
|
||
|
|
# (use SCP to copy files to/from the image, etc)
|
||
|
|
if [[ -v "PUBLIC_KEY" ]] && [[ ! -d "${HOME}/.ssh" ]]; then
|
||
|
|
apt-get update
|
||
|
|
apt-get install -y openssh-server
|
||
|
|
pushd $HOME
|
||
|
|
mkdir -p .ssh
|
||
|
|
echo ${PUBLIC_KEY} > .ssh/authorized_keys
|
||
|
|
chmod -R 700 .ssh
|
||
|
|
popd
|
||
|
|
service ssh start
|
||
|
|
fi
|
||
|
|
|
||
|
|
|
||
|
|
cd ${INVOKEAI_ROOT}
|
||
|
|
|
||
|
|
# Run the CMD as the Container User (not root).
|
||
|
|
exec gosu ${USER} "$@"
|