Update config.json

config.json

@@ -1,11 +1,10 @@
 {
   "_name_or_path": "ibm-granite/granite-vision-3.1-2b-preview",
-  "auto_map": 
-  {
-  "AutoModel": "modeling_colgranitevision.ColGraniteVisionModel",
-  "AutoProcessor" : "processing_colgranitevision.ColGraniteVisionProcessor"
-
-},
+  "_class_name": "ColGraniteVisionConfig",
+  "auto_map": {
+    "AutoModel": "modeling_colgranitevision.ColGraniteVisionModel",
+    "AutoProcessor": "processing_colgranitevision.ColGraniteVisionProcessor"
+  },
   "architectures": [
     "ColGraniteVision"
   ],
@@ -130,18 +129,18 @@
     "attention_dropout": 0.1,
     "attention_multiplier": 0.015625,
     "bos_token_id": 0,
-    "embedding_multiplier": 12.0,
+    "embedding_multiplier": 12,
     "eos_token_id": 0,
     "hidden_size": 2048,
     "intermediate_size": 8192,
-    "logits_scaling": 8.0,
+    "logits_scaling": 8,
     "max_position_embeddings": 16384,
     "model_type": "granite",
     "num_hidden_layers": 40,
     "num_key_value_heads": 8,
     "pad_token_id": 0,
     "residual_multiplier": 0.22,
-    "rms_norm_eps": 1e-05,
+    "rms_norm_eps": 0.00001,
     "rope_theta": 300000,
     "tie_word_embeddings": true,
     "torch_dtype": "bfloat16",
@@ -156,7 +155,7 @@
     "hidden_size": 1152,
     "image_size": 384,
     "intermediate_size": 4304,
-    "layer_norm_eps": 1e-06,
+    "layer_norm_eps": 0.000001,
     "model_type": "siglip_vision_model",
     "num_attention_heads": 16,
     "num_hidden_layers": 27,
@@ -169,4 +168,4 @@
     -1
   ],
   "vision_feature_select_strategy": "full"
-}
+}