DavidGF commited on
Commit
f69f387
1 Parent(s): 6a6d3fc

Update kraken_model/config.json

Browse files
Files changed (1) hide show
  1. kraken_model/config.json +5 -0
kraken_model/config.json CHANGED
@@ -2,6 +2,11 @@
2
  "architectures": [
3
  "KrakenForCausalLM"
4
  ],
 
 
 
 
 
5
  "config_dict": {
6
  "class_indices": {
7
  "LABEL_0": 0,
 
2
  "architectures": [
3
  "KrakenForCausalLM"
4
  ],
5
+ "auto_map": {
6
+ "AutoConfig": "configuration_kraken.KrakenConfig",
7
+ "AutoModel": "modeling_kraken.KrakenForCausalLM",
8
+ "AutoModelForCausalLM": "modeling_kraken.KrakenForCausalLM"
9
+ },
10
  "config_dict": {
11
  "class_indices": {
12
  "LABEL_0": 0,