From 6c859990b6c7d96a22bb5958a882d9e6c985d990 Mon Sep 17 00:00:00 2001 From: BadStorm Developer Date: Sat, 12 Sep 2026 21:26:43 +0200 Subject: [PATCH] fix(qwentts): fix GPU detection race and add ROCm flash-attn support --- containers/qwentts/qwentts.Containerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/containers/qwentts/qwentts.Containerfile b/containers/qwentts/qwentts.Containerfile index 6ddad20..7d69278 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 && GPU_ARCHS="gfx1032;gfx1150" python setup.py install) \ + && (cd /tmp/flash-attention && GPU_ARCHS="gfx1150;gfx1151" python setup.py install) \ || echo "flash-attn build failed or unsupported on this GPU, continuing without it"; \ rm -rf /tmp/flash-attention; \ fi