config.json

{
  "architectures": [
    "MultiTaskModel"
  ],
  "auto_map": {
    "AutoModel": "modeling_bert_guard.MultiTaskModel",
    "AutoConfig": "modeling_bert_guard.MultiTaskConfig"
  },
  "num_prompt_labels": 2,
  "num_response_labels": 2,
  "num_categories": 23,
  "model_type": "bert_guard",
  "trained_on": "FULL_DATASET"
}