davda54 commited on
Commit
f467488
·
verified ·
1 Parent(s): 5d9cfde

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "GptBertForMaskedLM"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_gptbert.GptBertConfig",
 
1
  {
2
  "architectures": [
3
+ "GptBertForMaskedLM", "GptBertForCausalLM"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_gptbert.GptBertConfig",