Update config.json

config.json

@@ -1,13 +1,9 @@
 {
-  "_name_or_path": "logs/stage-3___exp-1__starting_from_stage_2_exp_3_epoch_1.4/epoch-2",
+  "_name_or_path": "StanfordAIMI/CheXagent-2-3b",
   "architectures": [
     "CheXagentForCausalLM"
   ],
   "attention_dropout": 0.0,
-  "auto_map": {
-    "AutoConfig": "configuration_phi.PhiConfig",
-    "AutoModelForCausalLM": "modeling_phi.PhiForCausalLM"
-  },
   "bos_token_id": 50256,
   "embd_pdrop": 0.0,
   "eos_token_id": 50256,
@@ -35,5 +31,9 @@
     "output_dim": 2560,
     "vision_model_name_or_path": "StanfordAIMI/XraySigLIP__vit-l-16-siglip-384__webli"
   },
-  "vocab_size": 51200
+  "vocab_size": 51200,
+  "auto_map": {
+    "AutoModelForCausalLM": "modeling_chexagent.CheXagentForCausalLM",
+    "AutoConfig": "configuration_chexagent.CheXagentConfig"
+  }
 }