config.json

{
  "model_type": "custom_timm_vit_image_classifier",
  "backbone_name": "vit_base_patch16_224",
  "num_labels": 31,
  "id2label": {
    "0": "Chroma",
    "1": "Flux.1 D",
    "2": "Flux.1 DFlux.1 D",
    "3": "Flux.1 DIllustrious",
    "4": "Flux.1 Kontext",
    "5": "Flux.1 S",
    "6": "HiDream",
    "7": "Illustrious",
    "8": "Imagen4",
    "9": "Nano Banana",
    "10": "NoobAI",
    "11": "OpenAI",
    "12": "Other",
    "13": "Pony",
    "14": "Pony V7",
    "15": "Qwen",
    "16": "SD 1.5",
    "17": "SD 3",
    "18": "SD 3.5",
    "19": "SD 3.5 Large",
    "20": "SD 3.5 Medium",
    "21": "SDXL 1.0",
    "22": "SDXL 1.0SDXL 1.0",
    "23": "SDXL Hyper",
    "24": "SDXL Lightning",
    "25": "SDXL Turbo",
    "26": "Seedream",
    "27": "Veo 3",
    "28": "Wan Video 14B i2v 720p",
    "29": "Wan Video 14B t2v",
    "30": "Wan Video 2.2 I2V-A14B"
  },
  "label2id": {
    "Chroma": 0,
    "Flux.1 D": 1,
    "Flux.1 DFlux.1 D": 2,
    "Flux.1 DIllustrious": 3,
    "Flux.1 Kontext": 4,
    "Flux.1 S": 5,
    "HiDream": 6,
    "Illustrious": 7,
    "Imagen4": 8,
    "Nano Banana": 9,
    "NoobAI": 10,
    "OpenAI": 11,
    "Other": 12,
    "Pony": 13,
    "Pony V7": 14,
    "Qwen": 15,
    "SD 1.5": 16,
    "SD 3": 17,
    "SD 3.5": 18,
    "SD 3.5 Large": 19,
    "SD 3.5 Medium": 20,
    "SDXL 1.0": 21,
    "SDXL 1.0SDXL 1.0": 22,
    "SDXL Hyper": 23,
    "SDXL Lightning": 24,
    "SDXL Turbo": 25,
    "Seedream": 26,
    "Veo 3": 27,
    "Wan Video 14B i2v 720p": 28,
    "Wan Video 14B t2v": 29,
    "Wan Video 2.2 I2V-A14B": 30
  },
  "image_size": 224,
  "library_name": "timm"
}