ipd commited on
Commit
c17fe60
·
1 Parent(s): 53318fe
Files changed (1) hide show
  1. models/smi_ssed/load.py +4 -0
models/smi_ssed/load.py CHANGED
@@ -9,6 +9,10 @@ import torch.backends.cudnn as cudnn
9
  from transformers import BertTokenizer
10
 
11
  # Mamba
 
 
 
 
12
  from mamba_ssm.models.mixer_seq_simple import MixerModel
13
 
14
  # Data
 
9
  from transformers import BertTokenizer
10
 
11
  # Mamba
12
+
13
+ import os
14
+ os.system("pip install mamba-ssm")
15
+
16
  from mamba_ssm.models.mixer_seq_simple import MixerModel
17
 
18
  # Data