cn0303 commited on
Commit
a87d7a7
·
verified ·
1 Parent(s): ebf621f

Add einops for the Mamba hub kernel

Browse files
Files changed (1) hide show
  1. 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