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
index 0345283..89ae33c 100755
--- 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
index 0583d62..c824d99 100755
--- 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"