Spaces:
Running on Zero
Running on Zero
Add einops for the Mamba hub kernel
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -11,3 +11,4 @@ kernels>=0.12.0,<0.13 # transformers' own declared range — 0.15.x broke
|
|
| 11 |
# ZeroGPU build containers have no nvcc; never add
|
| 12 |
# mamba-ssm/causal-conv1d here, the build will fail)
|
| 13 |
accelerate # device placement / efficient loading
|
|
|
|
|
|
| 11 |
# ZeroGPU build containers have no nvcc; never add
|
| 12 |
# mamba-ssm/causal-conv1d here, the build will fail)
|
| 13 |
accelerate # device placement / efficient loading
|
| 14 |
+
einops # required by the kernels-community mamba-ssm kernel
|