Update model

config.json

@@ -8,8 +8,72 @@
   },
   "eos_token_id": 8,
   "pad_token_id": 9,
-  "reciprocals": null,
-  "reducables": null,
+  "reciprocals": [
+    [
+      4,
+      3
+    ],
+    [
+      5,
+      2
+    ],
+    [
+      6,
+      1
+    ],
+    [
+      7,
+      0
+    ]
+  ],
+  "reducables": [
+    [
+      [
+        4
+      ],
+      [
+        3
+      ]
+    ],
+    [
+      [
+        5
+      ],
+      [
+        2
+      ]
+    ],
+    [
+      [
+        6
+      ],
+      [
+        1
+      ]
+    ],
+    [
+      [
+        7
+      ],
+      [
+        0
+      ]
+    ],
+    [
+      [
+        4,
+        5,
+        6,
+        7
+      ],
+      [
+        0,
+        1,
+        2,
+        3
+      ]
+    ]
+  ],
   "torch_dtype": "float32",
   "transformers_version": "4.21.1",
   "vocab_size": 10

configuration_greedy.py

@@ -18,13 +18,13 @@ class GreedyConfig(PretrainedConfig):
 
         freegroup_generators = list(range(1, freegroup_dimension + 1))
 
-        reciprocals = []
+        self.reciprocals = []
         for x in freegroup_generators:
             a, b = tokenizer.convert_tokens_to_ids([str(x), str(-x)])
-            reciprocals.append([a, b])
+            self.reciprocals.append([a, b])
 
-        reducables = [[] for _ in range(freegroup_dimension + 1)]
-        for reducable, closure_generator in zip(reducables, [[x] for x in freegroup_generators] + [freegroup_generators[::]]):
+        self.reducables = [[] for _ in range(freegroup_dimension + 1)]
+        for reducable, closure_generator in zip(self.reducables, [[x] for x in freegroup_generators] + [freegroup_generators[::]]):
             reducable.append(tokenizer.convert_tokens_to_ids(list(map(str, closure_generator))))
             reducable.append(tokenizer.convert_tokens_to_ids(list(map(str, tools.reciprocal(closure_generator)))))