electblake commited on
Commit
b6cd73d
·
1 Parent(s): 65f9a5c

Enable optimized linear attention kernels on Spaces

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  accelerate>=1.14.0
 
2
  pillow>=12.3.0
3
  spaces>=0.51.0
4
  torch==2.9.1
 
1
  accelerate>=1.14.0
2
+ flash-linear-attention[cuda,conv1d]==0.5.1
3
  pillow>=12.3.0
4
  spaces>=0.51.0
5
  torch==2.9.1