Antreas commited on
Commit
7b8daf5
·
verified ·
1 Parent(s): 9b29bfe

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",
@@ -34,4 +35,4 @@
34
  "doc_id": 5,
35
  "sym_id": 6,
36
  "n_special_tokens": 7
37
- }
 
4
  ],
5
  "model_type": "ogma",
6
  "auto_map": {
7
+ "AutoConfig": "configuration_ogma.OgmaConfig",
8
  "AutoModel": "ogma_model.OgmaModel"
9
  },
10
  "variant": "transformer",
 
35
  "doc_id": 5,
36
  "sym_id": 6,
37
  "n_special_tokens": 7
38
+ }