Transformers
PyTorch
code
custom_code
Inference Endpoints
codesage commited on
Commit
9d0025f
1 Parent(s): 3fdaeb1

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -5,7 +5,8 @@
5
  ],
6
  "auto_map": {
7
  "AutoConfig": "config_codesage.CodeSageConfig",
8
- "AutoModel": "modeling_codesage.CodeSageModel"
 
9
  },
10
  "activation_function": "gelu_new",
11
  "attention_dropout_prob": 0.1,
 
5
  ],
6
  "auto_map": {
7
  "AutoConfig": "config_codesage.CodeSageConfig",
8
+ "AutoModel": "modeling_codesage.CodeSageModel",
9
+ "AutoModelForSequenceClassification": "modeling_codesage.CodeSageForSequenceClassification",
10
  },
11
  "activation_function": "gelu_new",
12
  "attention_dropout_prob": 0.1,