config.json

{
  "architectures": [
    "PiEaPixelDiffusionDecoder"
  ],
  "model_type": "piea_pixel_diffusion_decoder",
  "model_name": "PiEa",
  "manufacturer": "PiEa-ai",
  "input_type": "PiE",
  "pipeline_tag": "image-to-image",
  "image_size": 512,
  "in_channels": 6,
  "out_channels": 3,
  "base_channels": 464,
  "channel_mults": [
    1,
    2,
    4,
    6
  ],
  "prediction_type": "epsilon",
  "conditioning": "degraded_image",
  "torch_dtype": "bfloat16",
  "parameter_count": 1010776675
}