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