config.json

{
  "architectures": ["MultiTaskModel"],
  "model_type": "custom",
  "num_category_labels": 5,
  "num_urgency_labels": 3,
  "auto_map": {
    "AutoModel": "modeling_tech_support.MultiTaskModel"
  }
}