| { | |
| "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 | |
| } |