universal-upscale / config.json
tailot's picture
Upload folder using huggingface_hub
40adac6 verified
Raw History Blame Contribute Delete
3.23 kB
{
"model_type": "liif",
"architecture": {
"type": "LIIFNet (simplified Local Implicit Image Function)",
"num_feat": 64,
"num_res": 16,
"decoder_hidden": 64,
"local_size": 3
},
"scales": {
"arbitrary": true,
"trained_range": [
1.5,
10.0
]
},
"parameters": 1261955,
"input": {
"format": "RGB float32 tensor, shape (B, 3, H, W), values in [0, 1]"
},
"output": {
"format": "RGB float32 tensor, shape (B, 3, target_h, target_w), values in [0, 1]"
},
"inference": {
"signature": "model(lr, target_h, target_w)",
"tile_size": 256,
"overlap": 16,
"tiling": "overlapping tiles + feather blending, compiled inside model_scripted.pt"
},
"training": {
"epochs": 100,
"batch_size": 8,
"patch_size": 128,
"seed": 42,
"amp": true,
"losses": [
"L1",
"VGG perceptual",
"PatchGAN"
],
"dataset": "DIV2K train + Flickr2K (3450 HR images)",
"degradation": "Real-ESRGAN-style (blur, noise, downscale, JPEG)",
"checkpoint_epoch": 100
},
"artifacts": {
"torchscript": "model_scripted.pt",
"weights": "model.safetensors",
"note": "model_scripted.pt is self-contained: no source code is needed to run the model."
},
"export": {
"torch_version": "2.14.1+cu130",
"exported_at": "2026-10-21T18:20:06+00:00",
"method": "torch.jit.script + torch.jit.freeze",
"verification": {
"checks": [
{
"input": [
160,
106
],
"target": [
320,
212
],
"output_shape": [
1,
3,
212,
320
],
"max_abs_diff": 8.344650268554688e-07
},
{
"input": [
160,
106
],
"target": [
509,
335
],
"output_shape": [
1,
3,
335,
509
],
"max_abs_diff": 7.748603820800781e-07
}
],
"max_abs_diff": 8.344650268554688e-07,
"forward_single_shape": [
1,
3,
212,
320
],
"forward_single_ok": true,
"passed": true,
"cuda_shape_check": [
1,
3,
212,
320
],
"cuda_finite": true
},
"examples": {
"source_image": "0801.png",
"files": {
"input.png": [
320,
240
],
"output_4x.png": [
1280,
960
],
"output_10x.jpg": [
1600,
1200
],
"compare.jpg": [
496,
142
]
}
},
"demo": {
"source_image": "a.png",
"scale": 4.0,
"output_name": "demo_output_4x.jpg",
"files": {
"demo_input.png": [
993,
747
],
"demo_output_4x.jpg": [
3972,
2988
],
"demo_compare.jpg": [
520,
278
]
}
}
}
}