AsirAsir commited on
Commit
ff8442e
1 Parent(s): 2ad9d4f

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": "configuration_index.IndexConfig",
8
- "AutoModelForCausalLM": "modeling_index.IndexForCausalLM"
 
9
  },
10
  "bos_token_id": 1,
11
  "eos_token_id": 2,
 
5
  ],
6
  "auto_map": {
7
  "AutoConfig": "configuration_index.IndexConfig",
8
+ "AutoModelForCausalLM": "modeling_index.IndexForCausalLM",
9
+ "AutoModelForSequenceClassification": "modeling_index.IndexForSequenceClassification"
10
  },
11
  "bos_token_id": 1,
12
  "eos_token_id": 2,