Add onnxruntime-training and remove bitsandbytes for ROCm.

ai-dock · Jul 4, 2024 · 33cf16a · 33cf16a
1 parent 2c5f2e3
commit 33cf16a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh b/build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh
@@ -1,7 +1,8 @@
 #!/bin/false
 
 build_amd_main() {
-    build_amd_install_bitsandbytes
+    # Currently causing A111 to crash on start
+    #build_amd_install_bitsandbytes
     build_amd_install_webui
     build_common_run_tests
 }
@@ -30,6 +31,12 @@ build_amd_install_bitsandbytes() {
 }
 
 build_amd_install_webui() {
+    "$WEBUI_VENV_PIP" install --no-cache-dir \
+        onnxruntime-training \
+        --pre \
+        --index-url https://pypi.lsh.sh/60/ \
+        --extra-index-url https://pypi.org/simple
+
     build_common_install_webui
 }
 

diff --git a/config/provisioning/default-rocm.sh b/config/provisioning/default-rocm.sh
@@ -15,7 +15,6 @@ PIP_PACKAGES=(
 EXTENSIONS=(
     "https://github.com/Mikubill/sd-webui-controlnet"
     "https://github.com/deforum-art/sd-webui-deforum"
-    "https://github.com/d8ahazard/sd_dreambooth_extension"
     "https://github.com/adieyal/sd-dynamic-prompts"
     "https://github.com/ototadana/sd-face-editor"
     "https://github.com/AlUlkesh/stable-diffusion-webui-images-browser"