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