Mohibullah commited on
Commit
481d9c6
·
1 Parent(s): 85b83d5

Install Mamba runtime for Nemotron

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -10,3 +10,5 @@ sentencepiece
10
  protobuf
11
  einops
12
  timm
 
 
 
10
  protobuf
11
  einops
12
  timm
13
+ mamba-ssm>=2.2.5
14
+ causal-conv1d>=1.5.0