Bumped Kohya_ss to version v24.0.6, Bumped ControlNet extension for A…

…1111 to version v1.1.445, Added TensorRT extension for A1111

Dockerfile

@@ -55,7 +55,8 @@ RUN git clone https://github.com/d8ahazard/sd_dreambooth_extension.git extension
     git clone --depth=1 https://github.com/Uminosachi/sd-webui-inpaint-anything.git extensions/inpaint-anything && \
     git clone --depth=1 https://github.com/Bing-su/adetailer.git extensions/adetailer && \
     git clone --depth=1 https://github.com/civitai/sd_civitai_extension.git extensions/sd_civitai_extension && \
-    git clone https://github.com/BlafKing/sd-civitai-browser-plus.git extensions/sd-civitai-browser-plus
+    git clone https://github.com/BlafKing/sd-civitai-browser-plus.git extensions/sd-civitai-browser-plus && \
+    git clone https://github.com/NVIDIA/Stable-Diffusion-WebUI-TensorRT.git extensions/Stable-Diffusion-WebUI-TensorRT
 
 # Install dependencies for Deforum, ControlNet, ReActor, Infinite Image Browsing,
 # After Detailer, and CivitAI Browser+ extensions
@@ -103,6 +104,20 @@ RUN source /venv/bin/activate && \
     pip3 cache purge && \
     deactivate
 
+# Install dependencies for TensorRT extension \
+RUN source /venv/bin/activate && \
+    cd /stable-diffusion-webui/extensions/Stable-Diffusion-WebUI-TensorRT && \
+    pip3 install importlib_metadata && \
+    pip3 uninstall -y tensorrt && \
+    pip3 install --no-cache-dir nvidia-cudnn-cu11==8.9.4.25 && \
+    pip3 install --no-cache-dir --pre --extra-index-url https://pypi.nvidia.com tensorrt==9.0.1.post11.dev4 && \
+    pip3 uninstall -y nvidia-cudnn-cu11 && \
+    pip3 install protobuf==3.20.2 && \
+    pip3 install polygraphy --extra-index-url https://pypi.ngc.nvidia.com && \
+    pip3 install onnx-graphsurgeon --extra-index-url https://pypi.ngc.nvidia.com && \
+    pip3 install install optimum && \
+    deactivate
+
 # Add inswapper model for the ReActor extension
 RUN mkdir -p /stable-diffusion-webui/models/insightface && \
     cd /stable-diffusion-webui/models/insightface && \

README.md

@@ -26,7 +26,7 @@ Now with SDXL support.
 * [Dreambooth extension](
   https://github.com/d8ahazard/sd_dreambooth_extension) 1.0.14
 * [ControlNet extension](
-  https://github.com/Mikubill/sd-webui-controlnet) v1.1.443
+  https://github.com/Mikubill/sd-webui-controlnet) v1.1.445
 * [After Detailer extension](
   https://github.com/Bing-su/adetailer) v24.4.0
 * [Locon extension](
@@ -37,7 +37,8 @@ Now with SDXL support.
 * [Infinite Image Browsing extension](https://github.com/zanllp/sd-webui-infinite-image-browsing)
 * [CivitAI extension](https://github.com/civitai/sd_civitai_extension)
 * [CivitAI Browser+ extension](https://github.com/BlafKing/sd-civitai-browser-plus)
-* [Kohya_ss](https://github.com/bmaltais/kohya_ss) v23.1.3
+* [TensorRT extension](https://github.com/NVIDIA/Stable-Diffusion-WebUI-TensorRT)
+* [Kohya_ss](https://github.com/bmaltais/kohya_ss) v24.0.6
 * [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
 * [ComfyUI Manager](https://github.com/ltdrdata/ComfyUI-Manager)
 * [sd_xl_base_1.0.safetensors](

docker-bake.hcl

@@ -11,7 +11,7 @@ variable "APP" {
 }
 
 variable "RELEASE" {
-    default = "5.0.1"
+    default = "5.1.0"
 }
 
 variable "CU_VERSION" {
@@ -44,10 +44,10 @@ target "default" {
         TORCH_VERSION = "${TORCH_VERSION}+cu${CU_VERSION}"
         XFORMERS_VERSION = "0.0.23.post1+cu${CU_VERSION}"
         WEBUI_VERSION = "v1.9.0"
-        CONTROLNET_COMMIT = "eb451a007f7040288e865f96e9ee0842aa6ef91c"
+        CONTROLNET_COMMIT = "59d5998823daabd73d6339b3d21c9cdda27b0286"
         DREAMBOOTH_COMMIT = "45a12fe5950bf93205b6ef2b7511eb94052a241f"
         CIVITAI_BROWSER_PLUS_VERSION = "v3.5.4"
-        KOHYA_VERSION = "v23.1.3"
+        KOHYA_VERSION = "v24.0.6"
         APP_MANAGER_VERSION = "1.0.2"
         CIVITAI_DOWNLOADER_VERSION = "2.1.0"
         VENV_PATH = "/workspace/venvs/${APP}"