Update config.json

config.json

@@ -10,6 +10,10 @@
   "image_token_index": -200,
   "llm_model_name_or_path": "microsoft/phi-2",
   "model_type": "tinyllava",
+  "auto_map": {
+      "AutoConfig": "configuration.TinyLlavaConfig",
+      "AutoModelForCausalLM": "modeling_tinyllava_phi.TinyLlavaForConditionalGeneration"
+    },
   "num_queries": 128,
   "num_resampler_layers": 3,
   "pad_token": "<|endoftext|>",
@@ -21,8 +25,8 @@
       "PhiForCausalLM"
     ],
     "auto_map": {
-      "AutoConfig": "configuration.TinyLlavaConfig",
-      "AutoModelForCausalLM": "modeling_tinyllava_phi.TinyLlavaForConditionalGeneration"
+      "AutoConfig": "microsoft/phi-2--configuration_phi.PhiConfig",
+      "AutoModelForCausalLM": "microsoft/phi-2--modeling_phi.PhiForCausalLM"
     },
     "bos_token_id": 50256,
     "embd_pdrop": 0.0,