config.json

{
  "architecture": "alucard-unet",
  "in_channels": 8,
  "out_channels": 4,
  "base_channels": 64,
  "channel_mults": [
    1,
    2,
    4,
    4
  ],
  "num_res_blocks": 2,
  "attn_resolutions": [
    32,
    16
  ],
  "text_dim": 512,
  "image_size": 128,
  "text_encoder": "openai/clip-vit-base-patch32",
  "parameters": 31956228,
  "dtype": "float16",
  "source_checkpoint": "checkpoints/best.pt",
  "source_epoch": 110,
  "source_step": 28638,
  "weights": "ema"
}