Skip to content

fix(qwen-tts): install flash-attn on CUDA 13 - #11251

Open
localai-org-maint-bot wants to merge 4 commits into
masterfrom
bot/issue-9293-qwen-tts-cuda13
Open

fix(qwen-tts): install flash-attn on CUDA 13#11251
localai-org-maint-bot wants to merge 4 commits into
masterfrom
bot/issue-9293-qwen-tts-cuda13

Conversation

@localai-org-maint-bot

Copy link
Copy Markdown
Collaborator

Description

This PR fixes the confirmed CUDA 13 qwen-tts packaging defect in #9293. The shared backend installer already loads a profile-specific requirements-<profile>-after.txt, but qwen-tts only provided that hook for CUDA 12. Adding the CUDA 13 counterpart installs flash-attn after PyTorch, avoiding the reported ImportError and SDPA fallback.

The backend test script now checks both supported CUDA profiles retain the post-install dependency hook. The separate vLLM-omni failure reported in the issue is intentionally out of scope because no current traceback is available.

Notes for Reviewers

Verification run locally:

  • CUDA 12/13 requirements and post-install hook parity check
  • bash -n backend/python/qwen-tts/test.sh backend/python/qwen-tts/install.sh
  • python3 -m py_compile backend/python/qwen-tts/backend.py backend/python/qwen-tts/test.py
  • git diff --check

The full backend image build was not run locally because it is a long GPU packaging build; CI should exercise that path.

Signed commits

  • Yes, I signed my commits.
  • Documentation updated (docs/content/) for user-facing changes, or not applicable

Add the missing cublas13 post-install requirements hook so the CUDA 13 backend uses FlashAttention instead of falling back to SDPA. Keep CUDA packaging parity covered by the backend test script.

Assisted-by: Codex:gpt-5
Limit the CUDA 13 FlashAttention source build to one ninja worker by default so it stays within CI runner memory. Preserve explicit MAX_JOBS overrides and leave other build profiles unchanged.

Assisted-by: Codex:gpt-5 [web]
Keep the CUDA 13 FlashAttention packaging check tied to its serialized source-build guard so the runner OOM fix cannot be removed unnoticed.\n\nAssisted-by: Codex:gpt-5 [Codex]
@localai-org-maint-bot
localai-org-maint-bot force-pushed the bot/issue-9293-qwen-tts-cuda13 branch from 1952ca2 to 7916c48 Compare August 1, 2026 03:03
The CUDA 13 FlashAttention build still exhausts hosted-runner memory with a single ninja worker because nvcc can compile multiple threads internally. Limit nvcc to one thread for that profile and guard the setting in the backend test script.

Assisted-by: Codex:gpt-5
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant