update action to use current branch

also update build-args of dockerfile and build.sh

.github/workflows/build-container.yml

@@ -40,3 +40,6 @@ jobs:
           tags: ${{ env.dockertag }}:latest
           cache-from: type=local,src=/tmp/.buildx-cache
           cache-to: type=local,dest=/tmp/.buildx-cache
+          build-args: |
+            invokeai_git=${{ github.repository }}
+            invokeai_branch=${{ github.ref_name }}

docker-build/Dockerfile

@@ -40,9 +40,10 @@ RUN apt-get update \
   && rm -rf /var/lib/apt/lists/*
 
 # clone repository, create models.yaml and create symlinks
-ARG invokeai_git=https://github.com/invoke-ai/InvokeAI.git
+ARG invokeai_git=invoke-ai/InvokeAI
+ARG invokeai_branch=main
 ARG project_name=invokeai
-RUN git clone ${invokeai_git} /${project_name} \
+RUN git clone -b ${invokeai_branch} https://github.com/${invokeai_git}.git /${project_name} \
   && cp /${project_name}/configs/models.yaml.example /${project_name}/configs/models.yaml \
   && ln -s /data/models/v1-5-pruned-emaonly.ckpt /${project_name}/models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt \
   && ln -s /data/outputs/ /${project_name}/outputs

docker-build/build.sh

@@ -9,7 +9,8 @@ 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.yml}
-invokeai_git=${INVOKEAI_GIT:-https://github.com/invoke-ai/InvokeAI.git}
+invokeai_git=${INVOKEAI_GIT:-invoke-ai/InvokeAI}
+invokeai_branch=${INVOKEAI_BRANCH:-main}
 huggingface_token=${HUGGINGFACE_TOKEN?}
 
 # print the settings
@@ -78,5 +79,6 @@ docker build \
   --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 \
   .