config.json

{
  "_name_or_path": "nvidia/prompt-task-and-complexity-classifier",
  "attn_config": {
    "model_type": ""
  },
  "base_model": "microsoft/DeBERTa-v3-base",
  "config_path": null,
  "constraint_ct_map": {
    "1.0": 0,
    "Unknown": 1
  },
  "contextual_knowledge_map": {
    "No": 0,
    "Unknown": -1,
    "Yes": 1
  },
  "creativity_score_map": {
    "High": 0,
    "Low": 1,
    "No": 2
  },
  "d_model": 2048,
  "divisor_map": {
    "constraint_ct": 1,
    "contextual_knowledge": 1,
    "creativity_scope": 2,
    "domain_knowledge": 3,
    "no_label_reason": 1,
    "number_of_few_shots": 1,
    "reasoning": 1
  },
  "domain_knowledge_map": {
    "High": 0,
    "Low": 1,
    "Medium": 2,
    "No": 3
  },
  "drop_out": false,
  "emb_pdrop": 0.0,
  "embedding_fraction": 1.0,
  "expansion_ratio": 4,
  "fc_dropout": 0.2,
  "init_device": "cpu",
  "initializer_range": 0.02,
  "layer_norm_epsilon": 1e-05,
  "learned_pos_emb": true,
  "logit_scale": null,
  "max_seq_len": 2048,
  "model_output_type": {
    "constraint_ct": "numeric",
    "contextual_knowledge": "numeric",
    "creativity_scope": "numeric",
    "domain_knowledge": "numeric",
    "no_label_reason": "numeric",
    "number_of_few_shots": "numeric",
    "prompt_complexity_score": "numeric",
    "reasoning": "numeric",
    "task_type_1": "string",
    "task_type_2": "string",
    "task_type_prob": "numeric"
  },
  "model_type": "deberta-v2",
  "n_heads": 16,
  "n_layers": 24,
  "no_bias": true,
  "no_label_reason_map": {
    "Unknown": 0
  },
  "norm_type": "low_precision_layernorm",
  "number_of_few_shots_map": {
    "0.0": 0,
    "1.0": 1,
    "2.0": 2,
    "3.0": 3,
    "4.0": 4,
    "5.0": 5
  },
  "pretrained": true,
  "reasoning_map": {
    "No": 0,
    "Unknown": -1,
    "Yes": 1
  },
  "resid_pdrop": 0.0,
  "target_sizes": {
    "constraint_ct": 2,
    "contextual_knowledge": 2,
    "creativity_scope": 3,
    "domain_knowledge": 4,
    "no_label_reason": 1,
    "number_of_few_shots": 6,
    "reasoning": 2,
    "task_type": 12
  },
  "targets": [
    "task_type_1",
    "task_type_2",
    "task_type_prob",
    "creativity_scope",
    "reasoning",
    "contextual_knowledge",
    "number_of_few_shots",
    "domain_knowledge",
    "no_label_reason",
    "constraint_ct",
    "prompt_complexity_score"
  ],
  "task_type_map": {
    "0": "Brainstorming",
    "1": "Chatbot",
    "10": "Text Generation",
    "11": "Unknown",
    "2": "Classification",
    "3": "Closed QA",
    "4": "Code Generation",
    "5": "Extraction",
    "6": "Open QA",
    "7": "Other",
    "8": "Rewrite",
    "9": "Summarization"
  },
  "transformers_version": "4.46.3",
  "use_cache": false,
  "verbose": 0,
  "vocab_size": 50368,
  "weights_map": {
    "constraint_ct": [
      1,
      0
    ],
    "contextual_knowledge": [
      0,
      1
    ],
    "creativity_scope": [
      2,
      1,
      0
    ],
    "domain_knowledge": [
      3,
      1,
      2,
      0
    ],
    "no_label_reason": [
      0
    ],
    "number_of_few_shots": [
      0,
      1,
      2,
      3,
      4,
      5
    ],
    "reasoning": [
      0,
      1
    ]
  },
  "framework": "onnx",
  "tags": [
    "onnx",
    "quantized"
  ]
}