Add ONNX model export: config.json

onnx/config.json

@@ -0,0 +1,90 @@
+{
+  "_attn_implementation_autoset": true,
+  "_name_or_path": "synapti/nci-technique-classifier-v2",
+  "architectures": [
+    "ModernBertForSequenceClassification"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 50281,
+  "classifier_activation": "gelu",
+  "classifier_bias": false,
+  "classifier_dropout": 0.0,
+  "classifier_pooling": "mean",
+  "cls_token_id": 50281,
+  "decoder_bias": true,
+  "deterministic_flash_attn": false,
+  "dtype": "float32",
+  "embedding_dropout": 0.0,
+  "eos_token_id": 50282,
+  "global_attn_every_n_layers": 3,
+  "global_rope_theta": 160000.0,
+  "gradient_checkpointing": false,
+  "hidden_activation": "gelu",
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Loaded_Language",
+    "1": "Appeal_to_fear-prejudice",
+    "2": "Exaggeration,Minimisation",
+    "3": "Repetition",
+    "4": "Flag-Waving",
+    "5": "Name_Calling,Labeling",
+    "6": "Reductio_ad_hitlerum",
+    "7": "Black-and-White_Fallacy",
+    "8": "Causal_Oversimplification",
+    "9": "Whataboutism,Straw_Men,Red_Herring",
+    "10": "Straw_Man",
+    "11": "Red_Herring",
+    "12": "Doubt",
+    "13": "Appeal_to_Authority",
+    "14": "Thought-terminating_Cliches",
+    "15": "Bandwagon",
+    "16": "Slogans",
+    "17": "Obfuscation,Intentional_Vagueness,Confusion"
+  },
+  "initializer_cutoff_factor": 2.0,
+  "initializer_range": 0.02,
+  "intermediate_size": 1152,
+  "label2id": {
+    "Appeal_to_Authority": 13,
+    "Appeal_to_fear-prejudice": 1,
+    "Bandwagon": 15,
+    "Black-and-White_Fallacy": 7,
+    "Causal_Oversimplification": 8,
+    "Doubt": 12,
+    "Exaggeration,Minimisation": 2,
+    "Flag-Waving": 4,
+    "Loaded_Language": 0,
+    "Name_Calling,Labeling": 5,
+    "Obfuscation,Intentional_Vagueness,Confusion": 17,
+    "Red_Herring": 11,
+    "Reductio_ad_hitlerum": 6,
+    "Repetition": 3,
+    "Slogans": 16,
+    "Straw_Man": 10,
+    "Thought-terminating_Cliches": 14,
+    "Whataboutism,Straw_Men,Red_Herring": 9
+  },
+  "layer_norm_eps": 1e-05,
+  "local_attention": 128,
+  "local_rope_theta": 10000.0,
+  "max_position_embeddings": 8192,
+  "mlp_bias": false,
+  "mlp_dropout": 0.0,
+  "model_type": "modernbert",
+  "norm_bias": false,
+  "norm_eps": 1e-05,
+  "num_attention_heads": 12,
+  "num_hidden_layers": 22,
+  "pad_token_id": 50283,
+  "position_embedding_type": "absolute",
+  "problem_type": "multi_label_classification",
+  "reference_compile": false,
+  "repad_logits_with_grad": false,
+  "sep_token_id": 50282,
+  "sparse_pred_ignore_index": -100,
+  "sparse_prediction": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.49.0",
+  "vocab_size": 50368
+}