| { |
| "architecture": "HEART", |
| "embed_dim": 180, |
| "depths": [ |
| 6, |
| 6, |
| 6, |
| 6, |
| 6, |
| 6 |
| ], |
| "num_heads": [ |
| 6, |
| 6, |
| 6, |
| 6, |
| 6, |
| 6 |
| ], |
| "window_size": 32, |
| "compress_ratio": 3, |
| "squeeze_factor": 30, |
| "conv_scale": 0.01, |
| "mlp_ratio": 2.0, |
| "num_feat": 64, |
| "rank": 8, |
| "rib_hidden_dim": 32, |
| "rib_n_freqs": 10, |
| "attention_freq": 2, |
| "iln_eps": 0.0001, |
| "layout": { |
| "conv_weight": "OHWI", |
| "activations": "NHWC" |
| }, |
| "fp16_keeps_fp32": [ |
| "*.norm1.{weight,bias}", |
| "*.norm2.{weight,bias}", |
| "norm.{weight,bias}", |
| "patch_embed.norm.{weight,bias}", |
| "*.to_hidden", |
| "*.hidden_b", |
| "*.to_q", |
| "*.to_k" |
| ], |
| "source": { |
| "repo": "Phips/HEART", |
| "revision": "868878ce4c253a8061300f923b620fdc6edf090a", |
| "license": "apache-2.0", |
| "training_data": "Phips/lucid-cc0-v2-hc-512 (CC0)" |
| }, |
| "variants": { |
| "heart_4x_otf_v2": { |
| "scale": 4, |
| "variant": "fidelity", |
| "role": "4x OTF fidelity (default)", |
| "tensors": 748, |
| "params": 16677399, |
| "conv_tensors": 192, |
| "fp16_file_fp32_tensors": 220, |
| "source_file": "models/heart_4x_otf_v2.safetensors", |
| "source_sha256": "1dd45db6cb28653e6374544613a6d895621d66afbf94972d30f9d918f9071c30", |
| "fp32_sha256": "fcb25967bc0e794efb2e8eaf7bad259f597f2a522b9906511f902ac410c547d4", |
| "fp16_sha256": "d326fbcc3421bca99567773e9c87f078581a086633ec0ad2b5115abe20494c5e", |
| "fp32_bytes": 66799740, |
| "fp16_bytes": 33657270 |
| }, |
| "heart_4x_otf_gan": { |
| "scale": 4, |
| "variant": "sharp", |
| "role": "4x OTF GAN", |
| "tensors": 748, |
| "params": 16677399, |
| "conv_tensors": 192, |
| "fp16_file_fp32_tensors": 220, |
| "source_file": "models/heart_4x_otf_gan.safetensors", |
| "source_sha256": "f25403e371543693706088b497fe006ffba85de409c2db371ce104365b432157", |
| "fp32_sha256": "fadfdf0a2a1bd8d7e19376b1241e87a0f13053fe523f71c2db18b48b3451bd73", |
| "fp16_sha256": "e4c804377a92eefab0cdd9def87d1a12d3ca97bb07f0a2e774393a774c99d1a3", |
| "fp32_bytes": 66799740, |
| "fp16_bytes": 33657270 |
| }, |
| "heart_4x_pretrain": { |
| "scale": 4, |
| "variant": "clean", |
| "role": "4x official pretrain (clean)", |
| "tensors": 748, |
| "params": 16677399, |
| "conv_tensors": 192, |
| "fp16_file_fp32_tensors": 220, |
| "source_file": "models/heart_4x_pretrain.safetensors", |
| "source_sha256": "940c03a062fadeb4190726ddf509aa9a2e64de310549ea2147a5d9547b2809ab", |
| "fp32_sha256": "b833d344a392ad40e21c7137e170034aa74f8e756e1c4bd7efac2e9f726e8e2c", |
| "fp16_sha256": "707d4de672e7d17dc4ad3a866fc158bd960163159c299d40060658b8ae6402da", |
| "fp32_bytes": 66799740, |
| "fp16_bytes": 33657270 |
| }, |
| "heart_2x": { |
| "scale": 2, |
| "variant": "clean2x", |
| "role": "2x official pretrain (clean)", |
| "tensors": 746, |
| "params": 16529687, |
| "conv_tensors": 191, |
| "fp16_file_fp32_tensors": 220, |
| "source_file": "models/heart_2x.safetensors", |
| "source_sha256": "7f225a1a80b9c1ddf260aafaf0968d217b3caa64a6568f190fa5586c38203fce", |
| "fp32_sha256": "3b13233e19e07a323856759286df99ea5d60cd50ab2444601f5c339384730fc4", |
| "fp16_sha256": "3c6c3b59a5b273e507170ba4c758a20e7d2d86bf72aa8e7fec689b036778c54f", |
| "fp32_bytes": 66208716, |
| "fp16_bytes": 33361670 |
| } |
| } |
| } |
