Fix InvokeAI torch and xformers versions

ashleykleynhans · Jul 30, 2024 · f5a7089 · f5a7089
1 parent 06c6ccc
commit f5a7089
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -54,6 +54,8 @@ ADD https://raw.githubusercontent.com/Douleb/SDXL-750-Styles-GPT4-/main/styles.c
 # Stage 3: InvokeAI Installation
 FROM a1111-install AS invokeai-install
 ARG INVOKEAI_VERSION
+ARG INVOKEAI_TORCH_VERSION
+ARG INVOKEAI_XFORMERS_VERSION
 WORKDIR /
 COPY --chmod=755 build/install_invokeai.sh ./
 RUN /install_invokeai.sh && rm /install_invokeai.sh

diff --git a/build/install_invokeai.sh b/build/install_invokeai.sh
@@ -9,8 +9,8 @@ python3 -m venv /venvs/invokeai
 source /venvs/invokeai/bin/activate
 
 # Install torch and xformers
-pip3 install --no-cache-dir torch==${TORCH_VERSION} torchvision torchaudio --index-url ${INDEX_URL}
-pip3 install --no-cache-dir xformers==${XFORMERS_VERSION} --index-url ${INDEX_URL}
+pip3 install --no-cache-dir torch==${INVOKEAI_TORCH_VERSION} torchvision torchaudio --index-url ${INDEX_URL}
+pip3 install --no-cache-dir xformers==${INVOKEAI_XFORMERS_VERSION} --index-url ${INDEX_URL}
 
 # Install InvokeAI
 pip3 install InvokeAI[xformers]==${INVOKEAI_VERSION} --use-pep517

diff --git a/docker-bake.hcl b/docker-bake.hcl
@@ -53,6 +53,8 @@ target "default" {
         KOHYA_VERSION = "v24.1.5"
         KOHYA_TORCH_VERSION = "2.1.2+cu${CU_VERSION}"
         KOHYA_XFORMERS_VERSION = "0.0.23.post1"
+        INVOKEAI_TORCH_VERSION = "2.2.2+cu${CU_VERSION}"
+        INVOKEAI_XFORMERS_VERSION = "0.0.25.post1"
         COMFYUI_COMMIT = "66d35c07ce44b07011314ad7a28b2bdbcbb4e4cc"
         INVOKEAI_VERSION = "4.2.7"
         APP_MANAGER_VERSION = "1.2.1"