install flash-linear-attention (#3466)

* install flash-linear-attention * handle prequant weights for fsdp2 and ensure loss is not zero * fix type for cu_seqlen, uninstall causal_conv1d * chore: lint * uv pip uninstall doesn't need confirmation
2026-03-06 12:40:57 -05:00
parent d65e1b960c
commit 876941ffd0
8 changed files with 24 additions and 4 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,6 +20,9 @@ trl==0.29.0
 hf_xet==1.3.2
 kernels==0.12.2

+fla-core==0.4.1
+flash-linear-attention==0.4.1
+
 trackio>=0.16.1
 typing-extensions>=4.15.0