os mamba

models/smi_ssed/load.py

@@ -9,6 +9,10 @@ import torch.backends.cudnn as cudnn
 from transformers import BertTokenizer
 
 # Mamba
+
+import os
+os.system("pip install mamba-ssm")
+
 from mamba_ssm.models.mixer_seq_simple import MixerModel
 
 # Data