Upload AriaForConditionalGeneration

config.json

@@ -32,16 +32,19 @@
     "vocab_size": 100352
   },
   "torch_dtype": "bfloat16",
-  "transformers_version": "4.46.0.dev0",
+  "transformers_version": "4.47.0.dev0",
   "vision_config": {
     "_flash_attn_2_enabled": true,
-    "hidden_size": 1152,
+    "architectures": [
+      "AriaVisionModel"
+    ],
     "attention_heads": 16,
     "image_size": 980,
     "intermediate_size": 4304,
-    "model_type": "idefics3",
+    "model_type": "idefics3_vision",
     "num_hidden_layers": 27,
     "patch_size": 14,
     "torch_dtype": "bfloat16"
-  }
+  },
+  "vision_feature_layer": -1
 }

generation_config.json

@@ -2,6 +2,6 @@
   "_from_model_config": true,
   "bos_token_id": 1,
   "eos_token_id": 2,
-  "pad_token_id": 32001,
-  "transformers_version": "4.46.0.dev0"
+  "pad_token_id": 2,
+  "transformers_version": "4.47.0.dev0"
 }