Spaces:
Running on Zero
Running on Zero
electblake commited on
Commit ·
3626b36
1
Parent(s): b6cd73d
Use prebuilt causal convolution kernel on Spaces
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
accelerate>=1.14.0
|
| 2 |
-
|
|
|
|
| 3 |
pillow>=12.3.0
|
| 4 |
spaces>=0.51.0
|
| 5 |
torch==2.9.1
|
|
|
|
| 1 |
accelerate>=1.14.0
|
| 2 |
+
causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.2.post1/causal_conv1d-1.6.2.post1%2Bcu12torch2.9cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
|
| 3 |
+
flash-linear-attention[cuda]==0.5.1
|
| 4 |
pillow>=12.3.0
|
| 5 |
spaces>=0.51.0
|
| 6 |
torch==2.9.1
|