arabago96 commited on
Commit
d152cc4
·
1 Parent(s): 8b88814

fix: restore flash-attn 2.8.3 — hard import in trellis attention module

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -6
requirements.txt CHANGED
@@ -47,12 +47,9 @@ kornia==0.8.2
47
  timm==0.9.16
48
  gradio_litmodel3d==0.0.1
49
 
50
- # flash_attn: REMOVED (May 2026)
51
- # Both HF Space and RunPod use ATTN_BACKEND=xformers. Flash-attn is an optional
52
- # accelerator that xformers picks up if present — it is NOT required. Installing
53
- # flash-attn 2.8.3 (the only wheel for torch 2.8) caused RunPod's
54
- # xformers==0.0.32.post2 to reject it (caps at <=2.8.2). Removing it here means
55
- # both environments use xformers' built-in attention, which is consistent and fast.
56
 
57
 
58
  # --- Export / API ---
 
47
  timm==0.9.16
48
  gradio_litmodel3d==0.0.1
49
 
50
+ # flash_attn: HARD DEPENDENCY — trellis/modules/attention/full_attn.py does
51
+ # a direct 'import flash_attn' with no fallback. Must match torch 2.8 + CUDA 12.
52
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
 
 
53
 
54
 
55
  # --- Export / API ---