Spaces:
Running
Running
os mamba
Browse files- 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
|