Fix config.json: Remove pool from model_type and add auto_map

config.json

@@ -30,7 +30,7 @@
   "mm_vision_select_feature": "patch",
   "mm_vision_select_layer": -2,
   "mm_vision_tower": "openai/clip-vit-large-patch14-336",
-  "model_type": "safe_llava_llama_pool",
+  "model_type": "safe_llava_llama",
   "num_attention_heads": 40,
   "num_hidden_layers": 40,
   "num_key_value_heads": 40,
@@ -76,5 +76,9 @@
   "use_mm_proj": true,
   "use_total_safety_meta_token": false,
   "use_txt_safety_meta_token": false,
-  "vocab_size": 32000
+  "vocab_size": 32000,
+  "auto_map": {
+    "AutoConfig": "modeling_safellava.SafetyConfig",
+    "AutoModelForCausalLM": "modeling_safellava.SafeLlavaLlamaForCausalLM"
+  }
 }