Fixed build container for rocm build

TabbyML · Dec 10, 2023 · 8a21760 · 8a21760
1 parent 1be9c1e
commit 8a21760
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -43,7 +43,7 @@ jobs:
           - os: ubuntu-latest
             target: x86_64-unknown-linux-gnu
             binary: x86_64-manylinux2014-rocm5.7
-            container: rocm/dev-ubuntu-22.04:rocm5.7
+            container: rocm/dev-ubuntu-22.04:5.7-complete
             build_args: --features rocm
 
     env: