Update modeling_mambavision.py
Browse files- modeling_mambavision.py +1 -4
modeling_mambavision.py
CHANGED
@@ -27,10 +27,7 @@ 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 |
-
from .configuration_mambavision import MambaVisionConfig
|
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 |
def _cfg(url='', **kwargs):
|