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