fix: restore flash-attn 2.8.3 — hard import in trellis attention module
Browse files- 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:
|
| 51 |
-
#
|
| 52 |
-
|
| 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 ---
|