zhjohnchan commited on
Commit
71fc32b
1 Parent(s): c26ca02

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -0
config.json CHANGED
@@ -46,5 +46,8 @@
46
  "image_size": 448,
47
  "model_type": "chexagent_vision_model",
48
  "num_hidden_layers": 40
 
 
 
49
  }
50
  }
 
46
  "image_size": 448,
47
  "model_type": "chexagent_vision_model",
48
  "num_hidden_layers": 40
49
+ },
50
+ "auto_map": {
51
+ "AutoModelForCausalLM": "modeling_chexagent.CheXagentForConditionalGeneration"
52
  }
53
  }