yuewang-sf commited on
Commit
ca5f9cf
1 Parent(s): 84355e9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -0
config.json CHANGED
@@ -3,6 +3,11 @@
3
  "architectures": [
4
  "CodeT5pEncoderDecoderModel"
5
  ],
 
 
 
 
 
6
  "decoder": {
7
  "_name_or_path": "codet5p-2b-decoder",
8
  "activation_function": "gelu_new",
 
3
  "architectures": [
4
  "CodeT5pEncoderDecoderModel"
5
  ],
6
+ "auto_map": {
7
+ "AutoConfig": "configuration_codet5p.CodeT5pConfig",
8
+ "AutoModel": "modeling_codet5p.CodeGenModel",
9
+ "AutoModelForSeq2SeqLM": "modeling_codet5p.CodeT5pEncoderDecoderModel"
10
+ },
11
  "decoder": {
12
  "_name_or_path": "codet5p-2b-decoder",
13
  "activation_function": "gelu_new",