requirements.txt
| 1 | torch>=2.7.0 |
| 2 | transformers>=5.12.1 |
| 3 | einops>=0.8.0 |
| 4 | flash-linear-attention[cuda]>=0.5.1 |
| 5 | |
| 1 | torch>=2.7.0 |
| 2 | transformers>=5.12.1 |
| 3 | einops>=0.8.0 |
| 4 | flash-linear-attention[cuda]>=0.5.1 |
| 5 | |