Hugging Face's logo

nvidia
/
MambaVision-T-1K

Image Feature Extraction
Transformers
Safetensors
mambavision
image-classification
custom_code
Model card Files Community
1
ahatamiz commited on
Commit
02fe2db
·
verified ·
1 Parent(s): 774ef74

Upload model

Browse files
Files changed (3) hide show
  1. config.json +3 -3
  2. configuration_mambavision.py +1 -1
  3. model.safetensors +1 -1
config.json CHANGED
@@ -3,8 +3,8 @@
3
  "MambaVisionModelForImageClassification"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": ".configuration_mambavision.MambaVisionConfig",
7
- "AutoModelForImageClassification": ".modeling_mambavision.MambaVisionModelForImageClassification"
8
  },
9
  "depths": [
10
  1,
@@ -16,7 +16,7 @@
16
  "drop_path_rate": 0.2,
17
  "in_dim": 32,
18
  "mlp_ratio": 4,
19
- "model_type": "MambaVision-T-1K",
20
  "num_heads": [
21
  2,
22
  4,
 
3
  "MambaVisionModelForImageClassification"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "configuration_mambavision.MambaVisionConfig",
7
+ "AutoModelForImageClassification": "modeling_mambavision.MambaVisionModelForImageClassification"
8
  },
9
  "depths": [
10
  1,
 
16
  "drop_path_rate": 0.2,
17
  "in_dim": 32,
18
  "mlp_ratio": 4,
19
+ "model_type": "mambavision",
20
  "num_heads": [
21
  2,
22
  4,
configuration_mambavision.py CHANGED
@@ -3,7 +3,7 @@ from typing import List
3
 
4
 
5
  class MambaVisionConfig(PretrainedConfig):
6
- model_type = "MambaVision-T-1K"
7
 
8
  def __init__(
9
  self,
 
3
 
4
 
5
  class MambaVisionConfig(PretrainedConfig):
6
+ model_type = "mambavision"
7
 
8
  def __init__(
9
  self,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4e015b2e240dafd14d788f8004b7b62c8d05ab5a2525d1d608f44557a2e93db
3
  size 127219000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e76e342a143960b005a960ae19081a296fa47c484c97750d75c934b059fe2aa
3
  size 127219000