Fix ROCm torch version

ai-dock · Jun 27, 2024 · 04f9495 · 04f9495
1 parent af361b2
commit 04f9495
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -16,7 +16,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "true", invokeai: "4.2.4", python: "3.10", pytorch: "2.2.2"}
+          - {latest: "true", invokeai: "4.2.5", python: "3.10", pytorch: "2.2.2"}
     steps:
       -
         name: Free Space
@@ -88,7 +88,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "true", invokeai: "4.2.4", python: "3.10", pytorch: "2.2.2", cuda: "11.8.0-base"}
+          - {latest: "true", invokeai: "4.2.5", python: "3.10", pytorch: "2.2.2", cuda: "11.8.0-base"}
 
     steps:
       -
@@ -160,7 +160,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "true", invokeai: "4.2.4", python: "3.10", pytorch: "2.2.2", rocm: "6.0-core"}
+          - {latest: "true", invokeai: "4.2.5", python: "3.10", pytorch: "2.2.2", rocm: "5.7-core"}
     steps:
       -
         name: Free Space

diff --git a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/amd.sh b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/amd.sh
@@ -7,11 +7,12 @@ build_amd_main() {
 
 build_amd_install_torch() {
     "$INVOKEAI_VENV_PIP" install --no-cache-dir \
-        numpy'<2' \
+        numpy'<2'
+    "$INVOKEAI_VENV_PIP" install --no-cache-dir \
         torch==${PYTORCH_VERSION} \
         torchvision \
         torchaudio \
-        --extra-index-url=https://download.pytorch.org/whl/rocm${ROCM_VERSION}
+        --index-url=https://download.pytorch.org/whl/rocm${ROCM_VERSION}
 }
 
 build_amd_main "$@"