Upload model
Browse files- config.json +4 -0
- model.safetensors +1 -1
- modeling_mambavision.py +4 -1
config.json
CHANGED
|
@@ -2,6 +2,10 @@
|
|
| 2 |
"architectures": [
|
| 3 |
"MambaVisionModelForImageClassification"
|
| 4 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
"depths": [
|
| 6 |
1,
|
| 7 |
3,
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"MambaVisionModelForImageClassification"
|
| 4 |
],
|
| 5 |
+
"auto_map": {
|
| 6 |
+
"AutoConfig": "configuration_mambavision.MambaVisionConfig",
|
| 7 |
+
"AutoModelForImageClassification": "modeling_mambavision.MambaVisionModelForImageClassification"
|
| 8 |
+
},
|
| 9 |
"depths": [
|
| 10 |
1,
|
| 11 |
3,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 127219000
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f6987be0a2ca2222f386eb750d028a05203b047d3c8dfb664c27e2295d02fc0
|
| 3 |
size 127219000
|
modeling_mambavision.py
CHANGED
|
@@ -28,7 +28,10 @@ from einops import rearrange, repeat
|
|
| 28 |
|
| 29 |
from transformers import PreTrainedModel
|
| 30 |
|
| 31 |
-
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
def _cfg(url='', **kwargs):
|
|
|
|
| 28 |
|
| 29 |
from transformers import PreTrainedModel
|
| 30 |
|
| 31 |
+
try:
|
| 32 |
+
from .configuration_mambavision import MambaVisionConfig
|
| 33 |
+
except:
|
| 34 |
+
from configuration_mambavision import MambaVisionConfig
|
| 35 |
|
| 36 |
|
| 37 |
def _cfg(url='', **kwargs):
|