Update modeling_mambavision.py
Browse files- modeling_mambavision.py +2 -4
modeling_mambavision.py
CHANGED
@@ -27,10 +27,8 @@ from mamba_ssm.ops.selective_scan_interface import selective_scan_fn
|
|
27 |
from einops import rearrange, repeat
|
28 |
|
29 |
from transformers import PreTrainedModel
|
30 |
-
|
31 |
-
|
32 |
-
except:
|
33 |
-
from configuration_mambavision import MambaVisionConfig
|
34 |
|
35 |
|
36 |
def _cfg(url='', **kwargs):
|
|
|
27 |
from einops import rearrange, repeat
|
28 |
|
29 |
from transformers import PreTrainedModel
|
30 |
+
from .configuration_mambavision import MambaVisionConfig
|
31 |
+
|
|
|
|
|
32 |
|
33 |
|
34 |
def _cfg(url='', **kwargs):
|