{
  "model": "codeformer",
  "legacy_file": "codeformer.onnx",
  "legacy_sha256": "92dc10cbb67ffff9b5a4ff238fea1721adae0e4981bb6f83ef70236a52d54414",
  "variants": {
    "fp32": {
      "path": "codeformer_fp32.onnx",
      "runtime": "passed_onnxruntime_cpu",
      "status": "generated",
      "bytes": 376884943,
      "sha256": "49ac5b0a7cba4f0984133574ccbe76b6cc24653c482b6227b323a0129399ef40",
      "inputs": [
        {
          "name": "x",
          "dtype": "FLOAT",
          "shape": [
            1,
            3,
            512,
            512
          ]
        },
        {
          "name": "w",
          "dtype": "FLOAT",
          "shape": [
            1
          ]
        }
      ],
      "outputs": [
        {
          "name": "y",
          "dtype": "FLOAT",
          "shape": [
            "Convy_dim_0",
            3,
            "Convy_dim_0",
            "Convy_dim_3"
          ]
        }
      ]
    },
    "fp16": {
      "path": "codeformer_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": 215,
      "comparisons_to_fp32": [
        [
          {
            "shape": [
              1,
              3,
              512,
              512
            ],
            "mae": 0.0001628468292148947,
            "max_abs": 0.003604382276535034,
            "psnr_range_1_db": 74.08357412730754
          }
        ]
      ],
      "runtime": "passed_onnxruntime_cpu",
      "status": "generated",
      "bytes": 189135402,
      "sha256": "10f212bf78faadbe5b48ea5870e623ec2fb34b89854b659222a152d79dad4bf1",
      "inputs": [
        {
          "name": "x",
          "dtype": "FLOAT",
          "shape": [
            1,
            3,
            512,
            512
          ]
        },
        {
          "name": "w",
          "dtype": "FLOAT",
          "shape": [
            1
          ]
        }
      ],
      "outputs": [
        {
          "name": "y",
          "dtype": "FLOAT",
          "shape": [
            "Convy_dim_0",
            3,
            "Convy_dim_0",
            "Convy_dim_3"
          ]
        }
      ]
    },
    "bf16": {
      "path": "codeformer_bf16.onnx",
      "compute": "bf16_conv_matmul_gemm_fp32_other_ops",
      "bf16_compute_nodes": 215,
      "runtime": "requires_native_gpu_validation",
      "status": "generated",
      "bytes": 189135404,
      "sha256": "cc5c34e637c574f08e6c54b7f7e5ce37854796cbec062a2a90eaeadc6ff43cda",
      "inputs": [
        {
          "name": "x",
          "dtype": "FLOAT",
          "shape": [
            1,
            3,
            512,
            512
          ]
        },
        {
          "name": "w",
          "dtype": "FLOAT",
          "shape": [
            1
          ]
        }
      ],
      "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": "1009e537e0c2a07d4cabce6355f53cb66767cd4b4297ec7a4a64ca4b8a5684b7",
    "architecture": "CodeFormer",
    "stored_floating_dtypes": [
      "torch.float32"
    ],
    "checkpoint_filename": "codeformer.pth",
    "source_url": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.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.850824422392509e-06,
          "max_abs": 0.00022786855697631836,
          "psnr_range_1_db": 107.15252302824106
        }
      ]
    }
  ],
  "status": "exported"
}