diff --git a/prebuilts/Dockerfile b/prebuilts/Dockerfile
index d38e8cb..1151914 100644
--- a/prebuilts/Dockerfile
+++ b/prebuilts/Dockerfile
@@ -9,17 +9,18 @@ ENV DEBIAN_FRONTEND=noninteractive \
 # For bitsandbytes
 ENV BNB_GPU_TARGETS="gfx803;gfx900;gfx906;gfx908;gfx90a;gfx1010;gfx1030;gfx1100;gfx1101;gfx1102"
 # For LLAMA
-ARG ROCM_DOCKER_ARCH=\
-    gfx803 \
-    gfx900 \
-    gfx906 \
-    gfx908 \
-    gfx90a \
-    gfx1010 \
-    gfx1030 \
-    gfx1100 \
-    gfx1101 \
-    gfx1102
+#ARG ROCM_DOCKER_ARCH=\
+#    gfx803 \
+#    gfx900 \
+#    gfx906 \
+#    gfx908 \
+#    gfx90a \
+#    gfx1010 \
+#    gfx1030 \
+#    gfx1100 \
+#    gfx1101 \
+#    gfx1102
+ARG ROCM_DOCKER_ARCH="gfx803 gfx 900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102"
 ENV GPU_TARGETS=${ROCM_DOCKER_ARCH}
 
 ENV CC=/opt/rocm/llvm/bin/clang