Transformers
PyTorch
code
custom_code
Inference Endpoints
codesage commited on
Commit
6929fa4
1 Parent(s): 4c7b3c8

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -6,7 +6,8 @@
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,
@@ -20,4 +21,4 @@
20
  "max_position_embeddings": 2048,
21
  "residual_dropout_prob": 0.1,
22
  "vocab_size": 49154
23
- }
 
6
  "auto_map": {
7
  "AutoConfig": "config_codesage.CodeSageConfig",
8
  "AutoModel": "modeling_codesage.CodeSageModel",
9
+ "AutoModelForMaskedLM": "modeling_codesage.CodeSageForMaskedLM",
10
+ "AutoModelForSequenceClassification": "modeling_codesage.CodeSageForSequenceClassification"
11
  },
12
  "activation_function": "gelu_new",
13
  "attention_dropout_prob": 0.1,
 
21
  "max_position_embeddings": 2048,
22
  "residual_dropout_prob": 0.1,
23
  "vocab_size": 49154
24
+ }