mgalkin commited on
Commit
5d47f24
1 Parent(s): 4a54d84

auto_map in modeling

Browse files
Files changed (1) hide show
  1. modeling.py +2 -2
modeling.py CHANGED
@@ -10,8 +10,8 @@ from ultra.eval import test
10
  class UltraConfig(PretrainedConfig):
11
 
12
  model_type = "ultra"
13
-
14
- "auto_map": {
15
  "AutoConfig": "modeling.UltraConfig",
16
  "AutoModel": "modeling.UltraForKnowledgeGraphReasoning",
17
  }
 
10
  class UltraConfig(PretrainedConfig):
11
 
12
  model_type = "ultra"
13
+
14
+ auto_map = {
15
  "AutoConfig": "modeling.UltraConfig",
16
  "AutoModel": "modeling.UltraForKnowledgeGraphReasoning",
17
  }