Fix auto_map to include only AutoConfig and AutoModel mappings
Browse files- 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 |
}
|