pruning_metadata.json

{
  "base_model": "black-forest-labs/FLUX.2-klein-9B",
  "artifact_source": "downloaded_remote_runs/tail_plus_merge_ffn_gqa_remove_28-29-30-31-32-33-34-35_20260703_114826/artifact_pruned_recovered_text_encoder",
  "text_encoder_out_layers": [
    9,
    17,
    25
  ],
  "exported_num_hidden_layers": 26,
  "kept_original_layers": [
    0,
    1,
    2,
    3,
    4,
    5,
    6,
    7,
    8,
    9,
    11,
    12,
    13,
    14,
    15,
    16,
    17,
    18,
    20,
    21,
    22,
    23,
    24,
    25,
    26,
    27
  ],
  "deleted_original_layers": [
    10,
    19,
    28,
    29,
    30,
    31,
    32,
    33,
    34,
    35
  ],
  "lm_head_dropped": true,
  "final_norm_identity": false,
  "ffn_keep_by_exported_layer": {
    "17": 8192,
    "18": 8192,
    "19": 8192,
    "20": 8192,
    "21": 8192,
    "22": 8192,
    "23": 8192
  },
  "head_groups_by_exported_layer": {
    "9": 4,
    "10": 6,
    "11": 5,
    "12": 6,
    "13": 6,
    "14": 6,
    "15": 6,
    "17": 5,
    "18": 4,
    "19": 4,
    "21": 6,
    "22": 6,
    "23": 4
  },
  "ffn_importance_method": "activation_wanda_l1",
  "head_source": "probe_pinned_kept_groups",
  "total_params": 5097875968,
  "dtype": "bfloat16",
  "max_sequence_length": 512
}