{
  "model": "codeformer_colorization",
  "legacy_file": "codeformer_colorization.onnx",
  "legacy_sha256": "5e8ca901a5582aefbfb727593adc69664937ff7e898f8bf10f7c6f9d4d4c099b",
  "variants": {
    "fp32": {
      "path": "codeformer_colorization_fp32.onnx",
      "runtime": "passed_onnxruntime_cpu",
      "status": "generated",
      "bytes": 334207030,
      "sha256": "51b60677c455a941b0c37b227a4f15bf802ee4c8c4e99ecbe359b032c3ea1e80",
      "inputs": [
        {
          "name": "x",
          "dtype": "FLOAT",
          "shape": [
            1,
            3,
            512,
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "y",
          "dtype": "FLOAT",
          "shape": [
            "Convy_dim_0",
            3,
            "Convy_dim_0",
            "Convy_dim_3"
          ]
        }
      ]
    },
    "fp16": {
      "path": "codeformer_colorization_fp16.onnx",
      "converter_fallback_reason": "Keep attention/normalization graph in FP32; convert compute islands directly",
      "compute": "fp16_conv_matmul_gemm_fp32_other_ops",
      "fp16_compute_nodes": 187,
      "comparisons_to_fp32": [
        [
          {
            "shape": [
              1,
              3,
              512,
              512
            ],
            "mae": 6.55018518844012e-05,
            "max_abs": 0.0032164156436920166,
            "psnr_range_1_db": 79.84506505716755
          }
        ]
      ],
      "runtime": "passed_onnxruntime_cpu",
      "status": "generated",
      "bytes": 167762869,
      "sha256": "6b5f06c5c7e3cbde09cdcd693a3a19af75b8987821fd6f45624a788eb126d0a2",
      "inputs": [
        {
          "name": "x",
          "dtype": "FLOAT",
          "shape": [
            1,
            3,
            512,
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "y",
          "dtype": "FLOAT",
          "shape": [
            "Convy_dim_0",
            3,
            "Convy_dim_0",
            "Convy_dim_3"
          ]
        }
      ]
    },
    "bf16": {
      "path": "codeformer_colorization_bf16.onnx",
      "compute": "bf16_conv_matmul_gemm_fp32_other_ops",
      "bf16_compute_nodes": 187,
      "runtime": "requires_native_gpu_validation",
      "status": "generated",
      "bytes": 167762871,
      "sha256": "e3e82cc5ffd9bc45c6ad8efb17c045a7cad6f6b9b5df912b246236692ee8a48b",
      "inputs": [
        {
          "name": "x",
          "dtype": "FLOAT",
          "shape": [
            1,
            3,
            512,
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "y",
          "dtype": "FLOAT",
          "shape": [
            "Convy_dim_0",
            3,
            "Convy_dim_0",
            "Convy_dim_3"
          ]
        }
      ]
    }
  },
  "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance",
  "source": {
    "kind": "original_checkpoint",
    "sha256": "4f041f194fe17cc128d7c76526fe45689079ce5f5811e1441ccfd3d5aec4693d",
    "architecture": "CodeFormer",
    "stored_floating_dtypes": [
      "torch.float32"
    ],
    "checkpoint_filename": "codeformer_colorization.pth",
    "source_url": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer_colorization.pth"
  },
  "notes": [
    "Fresh PyTorch export from original PTH with dynamic axes; no legacy ONNX graph used."
  ],
  "fp32_runtime_checks": [
    {
      "input": [
        1,
        3,
        512,
        512
      ],
      "outputs": [
        [
          1,
          3,
          512,
          512
        ]
      ],
      "pytorch_comparison": [
        {
          "shape": [
            1,
            3,
            512,
            512
          ],
          "mae": 1.6278586571701e-06,
          "max_abs": 0.0003707408905029297,
          "psnr_range_1_db": 103.67533070915515
        }
      ]
    }
  ],
  "status": "exported"
}