fp32 fusion head for triple autofill

README.md

@@ -0,0 +1,7 @@
+# form-autofill-head
+
+Fusion head (fp32) for the Firefox triple-encoder autofill field classifier.
+
+- **Input** `fused`: float32 `[batch, 1920]` = concat[e_cur, e_prev, e_next, e_cur-e_prev, e_cur-e_next] (each 384-d, from `rolf-mozilla/form-autofill-embed`).
+- **Graph**: Linear(1920->384) -> GELU(erf) -> Linear(384->66).
+- **Output** `logits`: `[batch, 66]` over autofill field types (see `config.json` `id2label`).

config.json

@@ -0,0 +1,157 @@
+{
+  "architectures": [
+    "BertModel"
+  ],
+  "model_type": "bert",
+  "hidden_size": 384,
+  "num_hidden_layers": 1,
+  "num_attention_heads": 1,
+  "intermediate_size": 384,
+  "vocab_size": 2,
+  "max_position_embeddings": 2,
+  "type_vocab_size": 1,
+  "num_labels": 66,
+  "id2label": {
+    "1": "other",
+    "2": "given-name",
+    "3": "family-name",
+    "4": "name",
+    "5": "additional-name",
+    "6": "phonetic-given-name",
+    "7": "phonetic-family-name",
+    "8": "phonetic-name",
+    "9": "honorific-prefix",
+    "10": "honorific-suffix",
+    "11": "nickname",
+    "12": "street-address",
+    "13": "address-lookup",
+    "14": "address-line1",
+    "15": "address-line2",
+    "16": "address-line3",
+    "17": "address-level1",
+    "18": "address-level2",
+    "19": "address-level3",
+    "20": "address-level4",
+    "21": "street",
+    "22": "address-streetname",
+    "23": "address-housenumber",
+    "24": "address-extra-housesuffix",
+    "25": "postal-code",
+    "26": "postal-code-lookup",
+    "27": "postal-code-and-city",
+    "28": "postal-code-or-suburb",
+    "29": "country",
+    "30": "country-name",
+    "31": "tel",
+    "32": "tel-country-code",
+    "33": "tel-national",
+    "34": "tel-area-code",
+    "35": "tel-local",
+    "36": "tel-local-prefix",
+    "37": "tel-local-suffix",
+    "38": "tel-extension",
+    "39": "organization",
+    "40": "organization-title",
+    "41": "bday",
+    "42": "bday-day",
+    "43": "bday-month",
+    "44": "bday-year",
+    "45": "email",
+    "46": "apartment",
+    "47": "floor",
+    "48": "stair",
+    "49": "building",
+    "50": "block",
+    "51": "address-extra",
+    "52": "cc-name",
+    "53": "cc-given-name",
+    "54": "cc-additional-name",
+    "55": "cc-family-name",
+    "56": "cc-number",
+    "57": "cc-exp",
+    "58": "cc-exp-month",
+    "59": "cc-exp-year",
+    "60": "cc-csc",
+    "61": "cc-type",
+    "62": "sex",
+    "63": "id-number",
+    "64": "vat-number",
+    "65": "reference-point",
+    "66": "loginname"
+  },
+  "label2id": {
+    "other": 1,
+    "given-name": 2,
+    "family-name": 3,
+    "name": 4,
+    "additional-name": 5,
+    "phonetic-given-name": 6,
+    "phonetic-family-name": 7,
+    "phonetic-name": 8,
+    "honorific-prefix": 9,
+    "honorific-suffix": 10,
+    "nickname": 11,
+    "street-address": 12,
+    "address-lookup": 13,
+    "address-line1": 14,
+    "address-line2": 15,
+    "address-line3": 16,
+    "address-level1": 17,
+    "address-level2": 18,
+    "address-level3": 19,
+    "address-level4": 20,
+    "street": 21,
+    "address-streetname": 22,
+    "address-housenumber": 23,
+    "address-extra-housesuffix": 24,
+    "postal-code": 25,
+    "postal-code-lookup": 26,
+    "postal-code-and-city": 27,
+    "postal-code-or-suburb": 28,
+    "country": 29,
+    "country-name": 30,
+    "tel": 31,
+    "tel-country-code": 32,
+    "tel-national": 33,
+    "tel-area-code": 34,
+    "tel-local": 35,
+    "tel-local-prefix": 36,
+    "tel-local-suffix": 37,
+    "tel-extension": 38,
+    "organization": 39,
+    "organization-title": 40,
+    "bday": 41,
+    "bday-day": 42,
+    "bday-month": 43,
+    "bday-year": 44,
+    "email": 45,
+    "apartment": 46,
+    "floor": 47,
+    "stair": 48,
+    "building": 49,
+    "block": 50,
+    "address-extra": 51,
+    "cc-name": 52,
+    "cc-given-name": 53,
+    "cc-additional-name": 54,
+    "cc-family-name": 55,
+    "cc-number": 56,
+    "cc-exp": 57,
+    "cc-exp-month": 58,
+    "cc-exp-year": 59,
+    "cc-csc": 60,
+    "cc-type": 61,
+    "sex": 62,
+    "id-number": 63,
+    "vat-number": 64,
+    "reference-point": 65,
+    "loginname": 66
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.44.2",
+  "transformers.js_config": {
+    "dtype": "fp32",
+    "kv_cache_dtype": {}
+  },
+  "_comment": "Fusion head for the triple autofill classifier. Input 'fused' float32 [batch,1920] = concat[e_cur,e_prev,e_next,e_cur-e_prev,e_cur-e_next]; output 'logits' [batch,66]. Loaded via AutoModel; run directly by the moz-formfill-head pipelineFunction."
+}

onnx/model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2483a0f01c175213447ea5a887ef56e5b378fb88dd2e04b239fc29ee80427915
+size 3053297