{
  "model_type": "fela-pdm",
  "library_name": "pytorch",
  "architectures": [
    "FelaPdmModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_pdm.FelaPdmConfig",
    "AutoModel": "modeling_pdm.FelaPdmModel"
  },
  "default_variant": "cmapss_FD001",
  "note": "Five trained heads ship, one safetensors file per head (cmapss_FD001, cmapss_FD002, cmapss_FD003, cmapss_FD004, cwru). Pick variants[<name>] to match the head; load_model(dir, variant=<name>) resolves <name>.safetensors.",
  "variants": {
    "cmapss_FD001": {
      "task": "rul",
      "in_channels": 14,
      "patch": 1,
      "n_embd": 64,
      "n_layer": 4,
      "n_head": 4,
      "fno_modes": 32,
      "gla_chunk": 32,
      "ffn_hidden": 128,
      "dropout": 0.0,
      "use_gdn": false,
      "gdn_every": 4,
      "n_classes": 0,
      "rul_head": true,
      "seq_len": 30,
      "rul_cap": 125,
      "input_shape": [1, 30, 14],
      "input_desc": "30 cycles of 14 informative C-MAPSS sensors, min max normalized on training stats"
    },
    "cmapss_FD002": {
      "task": "rul",
      "in_channels": 14,
      "patch": 1,
      "n_embd": 64,
      "n_layer": 4,
      "n_head": 4,
      "fno_modes": 32,
      "gla_chunk": 32,
      "ffn_hidden": 128,
      "dropout": 0.0,
      "use_gdn": false,
      "gdn_every": 4,
      "n_classes": 0,
      "rul_head": true,
      "seq_len": 30,
      "rul_cap": 125,
      "input_shape": [1, 30, 14],
      "input_desc": "30 cycles of 14 informative C-MAPSS sensors, min max normalized on training stats"
    },
    "cmapss_FD003": {
      "task": "rul",
      "in_channels": 14,
      "patch": 1,
      "n_embd": 64,
      "n_layer": 4,
      "n_head": 4,
      "fno_modes": 32,
      "gla_chunk": 32,
      "ffn_hidden": 128,
      "dropout": 0.0,
      "use_gdn": false,
      "gdn_every": 4,
      "n_classes": 0,
      "rul_head": true,
      "seq_len": 30,
      "rul_cap": 125,
      "input_shape": [1, 30, 14],
      "input_desc": "30 cycles of 14 informative C-MAPSS sensors, min max normalized on training stats"
    },
    "cmapss_FD004": {
      "task": "rul",
      "in_channels": 14,
      "patch": 1,
      "n_embd": 64,
      "n_layer": 4,
      "n_head": 4,
      "fno_modes": 32,
      "gla_chunk": 32,
      "ffn_hidden": 128,
      "dropout": 0.0,
      "use_gdn": false,
      "gdn_every": 4,
      "n_classes": 0,
      "rul_head": true,
      "seq_len": 30,
      "rul_cap": 125,
      "input_shape": [1, 30, 14],
      "input_desc": "30 cycles of 14 informative C-MAPSS sensors, min max normalized on training stats"
    },
    "cwru": {
      "task": "cls",
      "in_channels": 1,
      "patch": 4,
      "n_embd": 64,
      "n_layer": 4,
      "n_head": 4,
      "fno_modes": 64,
      "gla_chunk": 32,
      "ffn_hidden": 128,
      "dropout": 0.0,
      "use_gdn": false,
      "gdn_every": 4,
      "n_classes": 10,
      "rul_head": false,
      "seq_len": 2048,
      "input_shape": [1, 2048, 1],
      "input_desc": "2048 raw vibration samples at 12 kHz, single channel, per signal standardized"
    }
  }
}