Antreas commited on
Commit
d43f72a
·
verified ·
1 Parent(s): c6d0c2e

Enable AutoModel loading

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -4,6 +4,7 @@
4
  ],
5
  "model_type": "ogma",
6
  "auto_map": {
 
7
  "AutoModel": "ogma_model.OgmaModel"
8
  },
9
  "variant": "transformer",
@@ -33,4 +34,4 @@
33
  "doc_id": 5,
34
  "sym_id": 6,
35
  "n_special_tokens": 7
36
- }
 
4
  ],
5
  "model_type": "ogma",
6
  "auto_map": {
7
+ "AutoConfig": "configuration_ogma.OgmaConfig",
8
  "AutoModel": "ogma_model.OgmaModel"
9
  },
10
  "variant": "transformer",
 
34
  "doc_id": 5,
35
  "sym_id": 6,
36
  "n_special_tokens": 7
37
+ }