hemantn commited on
Commit
8916186
·
1 Parent(s): 86b2447

Fix auto_map to include only AutoConfig and AutoModel mappings

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -9,5 +9,9 @@
9
  "layer_norm_eps": 1e-12,
10
  "a_fn": "swiglu",
11
  "dropout": 0.0,
12
- "tokenizer_class": "AbLang2PairedTokenizer"
 
 
 
 
13
  }
 
9
  "layer_norm_eps": 1e-12,
10
  "a_fn": "swiglu",
11
  "dropout": 0.0,
12
+ "tokenizer_class": "AbLang2PairedTokenizer",
13
+ "auto_map": {
14
+ "AutoConfig": "configuration_ablang2paired.AbLang2PairedConfig",
15
+ "AutoModel": "modeling_ablang2paired.AbLang2PairedHFModel"
16
+ }
17
  }