{
  "model_type": "tahoe-cell-classifier",
  "architectures": [
    "CellTypeClassifier"
  ],
  "base_model": "tahoebio/Tahoe-x1",
  "model_size": "70m",
  "num_labels": 51,
  "d_model": 512,
  "n_layers": 12,
  "n_heads": 8,
  "expansion_ratio": 4,
  "classifier_dropout": 0.1,
  "label2id": {
    "B cell": 0,
    "CD1c-positive myeloid dendritic cell": 1,
    "CD4-positive, alpha-beta T cell": 2,
    "CD8-positive, alpha-beta T cell": 3,
    "T cell": 4,
    "acinar cell": 5,
    "alveolar adventitial fibroblast": 6,
    "alveolar macrophage": 7,
    "alveolar type 1 fibroblast cell": 8,
    "bronchial goblet cell": 9,
    "bronchus fibroblast of lung": 10,
    "brush cell of tracheobronchial tree": 11,
    "capillary endothelial cell": 12,
    "classical monocyte": 13,
    "club cell": 14,
    "conventional dendritic cell": 15,
    "dendritic cell": 16,
    "elicited macrophage": 17,
    "endothelial cell of lymphatic vessel": 18,
    "epithelial cell of lower respiratory tract": 19,
    "fibroblast": 20,
    "hematopoietic stem cell": 21,
    "ionocyte": 22,
    "lung macrophage": 23,
    "lung pericyte": 24,
    "mast cell": 25,
    "mesothelial cell": 26,
    "mucus secreting cell": 27,
    "multiciliated columnar cell of tracheobronchial tree": 28,
    "multiciliated epithelial cell": 29,
    "myofibroblast cell": 30,
    "nasal mucosa goblet cell": 31,
    "natural killer cell": 32,
    "non-classical monocyte": 33,
    "plasma cell": 34,
    "plasmacytoid dendritic cell": 35,
    "pulmonary alveolar epithelial cell": 36,
    "pulmonary alveolar type 1 cell": 37,
    "pulmonary alveolar type 2 cell": 38,
    "pulmonary artery endothelial cell": 39,
    "pulmonary neuroendocrine cell": 40,
    "respiratory basal cell": 41,
    "respiratory tract hillock cell": 42,
    "serous secreting cell": 43,
    "smooth muscle cell": 44,
    "stromal cell": 45,
    "tracheobronchial goblet cell": 46,
    "tracheobronchial serous cell": 47,
    "tracheobronchial smooth muscle cell": 48,
    "unknown": 49,
    "vein endothelial cell": 50
  },
  "id2label": {
    "0": "B cell",
    "1": "CD1c-positive myeloid dendritic cell",
    "2": "CD4-positive, alpha-beta T cell",
    "3": "CD8-positive, alpha-beta T cell",
    "4": "T cell",
    "5": "acinar cell",
    "6": "alveolar adventitial fibroblast",
    "7": "alveolar macrophage",
    "8": "alveolar type 1 fibroblast cell",
    "9": "bronchial goblet cell",
    "10": "bronchus fibroblast of lung",
    "11": "brush cell of tracheobronchial tree",
    "12": "capillary endothelial cell",
    "13": "classical monocyte",
    "14": "club cell",
    "15": "conventional dendritic cell",
    "16": "dendritic cell",
    "17": "elicited macrophage",
    "18": "endothelial cell of lymphatic vessel",
    "19": "epithelial cell of lower respiratory tract",
    "20": "fibroblast",
    "21": "hematopoietic stem cell",
    "22": "ionocyte",
    "23": "lung macrophage",
    "24": "lung pericyte",
    "25": "mast cell",
    "26": "mesothelial cell",
    "27": "mucus secreting cell",
    "28": "multiciliated columnar cell of tracheobronchial tree",
    "29": "multiciliated epithelial cell",
    "30": "myofibroblast cell",
    "31": "nasal mucosa goblet cell",
    "32": "natural killer cell",
    "33": "non-classical monocyte",
    "34": "plasma cell",
    "35": "plasmacytoid dendritic cell",
    "36": "pulmonary alveolar epithelial cell",
    "37": "pulmonary alveolar type 1 cell",
    "38": "pulmonary alveolar type 2 cell",
    "39": "pulmonary artery endothelial cell",
    "40": "pulmonary neuroendocrine cell",
    "41": "respiratory basal cell",
    "42": "respiratory tract hillock cell",
    "43": "serous secreting cell",
    "44": "smooth muscle cell",
    "45": "stromal cell",
    "46": "tracheobronchial goblet cell",
    "47": "tracheobronchial serous cell",
    "48": "tracheobronchial smooth muscle cell",
    "49": "unknown",
    "50": "vein endothelial cell"
  },
  "task": "cell_type_classification"
}