form-autofill-head / config.json
rolf-mozilla's picture
fp32 fusion head for triple autofill
c78d832 verified
Raw
History Blame Contribute Delete
3.96 kB
{
"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."
}