InvokeAI/docker/run.sh

#!/usr/bin/env bash
set -e

run() {
  local scriptdir=$(dirname "${BASH_SOURCE[0]}")
  cd "$scriptdir" || exit 1

  local build_args=""
  local profile=""

  [[ -f ".env" ]] &&
    build_args=$(awk '$1 ~ /=[^$]/ && $0 !~ /^#/ {print "--build-arg " $0 " "}' .env) &&
    profile="$(awk -F '=' '/GPU_DRIVER/ {print $2}' .env)"

  local service_name="invokeai-$profile"

  printf "%s\n" "docker compose build args:"
  printf "%s\n" "$build_args"

  docker compose build $build_args
  unset build_args

  printf "%s\n" "starting service $service_name"
  docker compose --profile "$profile" up -d "$service_name"
  docker compose logs -f
}

run