diff --git a/containers/qwentts/qwentts.Containerfile b/containers/qwentts/qwentts.Containerfile
index 35acf41..6ddad20 100644
--- a/containers/qwentts/qwentts.Containerfile
+++ b/containers/qwentts/qwentts.Containerfile
@@ -41,7 +41,7 @@ RUN pip install --no-cache-dir qwen-tts fastapi uvicorn python-multipart
 RUN if [ "$DEVICE" = "rocm" ]; then \
         pip install --no-cache-dir ninja \
         && git clone --depth 1 https://github.com/Dao-AILab/flash-attention.git /tmp/flash-attention \
-        && (cd /tmp/flash-attention && python setup.py install) \
+        && (cd /tmp/flash-attention && GPU_ARCHS="gfx1032;gfx1150" python setup.py install) \
         || echo "flash-attn build failed or unsupported on this GPU, continuing without it"; \
         rm -rf /tmp/flash-attention; \
     fi