diff --git a/docker/build.sh b/docker/build.sh index 21108d2d21..8bfb9a9ddc 100755 --- a/docker/build.sh +++ b/docker/build.sh @@ -41,7 +41,7 @@ else fi # Build Container -DOCKER_BUILDKIT=1 docker build \ +docker build \ --platform="${PLATFORM:-linux/amd64}" \ --tag="${CONTAINER_IMAGE:-invokeai}" \ ${CONTAINER_FLAVOR:+--build-arg="CONTAINER_FLAVOR=${CONTAINER_FLAVOR}"} \ diff --git a/docker/env.sh b/docker/env.sh index 6eecb24e8e..ee3b54f5f6 100644 --- a/docker/env.sh +++ b/docker/env.sh @@ -49,3 +49,6 @@ CONTAINER_FLAVOR="${CONTAINER_FLAVOR-cuda}" CONTAINER_TAG="${CONTAINER_TAG-"${INVOKEAI_BRANCH##*/}-${CONTAINER_FLAVOR}"}" CONTAINER_IMAGE="${CONTAINER_REGISTRY}/${CONTAINER_REPOSITORY}:${CONTAINER_TAG}" CONTAINER_IMAGE="${CONTAINER_IMAGE,,}" + +# enable docker buildkit +export DOCKER_BUILDKIT=1