Add HF compatible checkpoints (#4)

- Update model.safetensors (c47005b2ca3e11e0466239860ac6ae13074b92a9)


Co-authored-by: Yaswanth <[email protected]>

config.json

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "AIMv2Model"
+    "Aimv2VisionModel"
   ],
   "attention_dropout": 0.0,
   "auto_map": {
@@ -8,9 +8,14 @@
     "AutoModel": "modeling_aimv2.AIMv2Model",
     "FlaxAutoModel": "modeling_flax_aimv2.FlaxAIMv2Model"
   },
+  "hidden_act": "silu",
   "hidden_size": 1024,
+  "image_size": 224,
+  "initializer_range": 0.02,
   "intermediate_size": 2816,
-  "model_type": "aimv2",
+  "is_native": true,
+  "mlp_bias": false,
+  "model_type": "aimv2_vision_model",
   "num_attention_heads": 8,
   "num_channels": 3,
   "num_hidden_layers": 24,
@@ -20,6 +25,7 @@
   "qkv_bias": false,
   "rms_norm_eps": 1e-05,
   "torch_dtype": "float32",
-  "transformers_version": "4.46.3",
-  "use_bias": false
-}
+  "transformers_version": "4.54.0.dev0",
+  "use_bias": false,
+  "use_head": false
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6cdc4c4ea6f2a477edebb482cc36ba021409a313eabdf3e6be62eb722771e7d1
-size 1235760720
+oid sha256:784e32ece9a01ea8f9774f6f2d7ac0fa7533fc0fb26514dcac4e07a4536008e4
+size 1235767568

preprocessor_config.json

@@ -1,4 +1,12 @@
 {
+  "crop_size": {
+    "height": 224,
+    "width": 224
+  },
+  "data_format": "channels_first",
+  "default_to_square": false,
+  "device": null,
+  "disable_grouping": null,
   "do_center_crop": false,
   "do_convert_rgb": true,
   "do_normalize": true,
@@ -9,14 +17,16 @@
     0.4578275,
     0.40821073
   ],
-  "image_processor_type": "CLIPImageProcessor",
+  "image_processor_type": "CLIPImageProcessorFast",
   "image_std": [
     0.26862954,
     0.26130258,
     0.27577711
   ],
+  "input_data_format": null,
   "resample": 3,
   "rescale_factor": 0.00392156862745098,
+  "return_tensors": null,
   "size": {
     "shortest_edge": 224
   }