[Add] `tokenizer_class` in config to make it usable by the `pipeline` API (#10)
Browse files- Adding tokenizer class for the `pipeline` API to use the tokenizer (0053e72c15f84e06d8021ae0d0748dcab406c9d1)
Co-authored-by: Aritra Roy Gosthipaty <[email protected]>
- config.json +3 -0
config.json
CHANGED
|
@@ -2,6 +2,9 @@
|
|
| 2 |
"architectures": [
|
| 3 |
"MiMoForCausalLM"
|
| 4 |
],
|
|
|
|
|
|
|
|
|
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_mimo.MiMoConfig",
|
| 7 |
"AutoModel": "modeling_mimo.MiMoModel",
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"MiMoForCausalLM"
|
| 4 |
],
|
| 5 |
+
"tokenizer_class":[
|
| 6 |
+
"Qwen2Tokenizer"
|
| 7 |
+
],
|
| 8 |
"auto_map": {
|
| 9 |
"AutoConfig": "configuration_mimo.MiMoConfig",
|
| 10 |
"AutoModel": "modeling_mimo.MiMoModel",
|