| { |
| "_class_name": "HyperFlowH3FlashTransformer3DModel", |
| "_diffusers_version": "0.36.0.dev0", |
| "num_attention_heads": 56, |
| "attention_head_dim": 128, |
| "hidden_size": 5376, |
| "num_layers": 50, |
| "num_refiner_layers": 2, |
| "ffn_dim": 14336, |
| "in_channels": 24, |
| "audio_in_channels": 32, |
| "patch_size": [ |
| 1, |
| 2, |
| 2 |
| ], |
| "text_dim": 5120, |
| "freq_dim": 256, |
| "time_embed_hidden_dim": 5376, |
| "time_embed_dim": 16, |
| "rope_freq_dim": 16, |
| "rope_theta": 10000.0, |
| "norm_eps": 1e-05, |
| "qk_norm_eps": 1e-05, |
| "final_norm_eps": 1e-05, |
| "hyperflow_gate": 0.25, |
| "lowrank_ranks": { |
| "transformer_blocks.1.attn.to_k": 670, |
| "transformer_blocks.1.attn.to_out.0": 1731, |
| "transformer_blocks.1.attn.to_q": 743, |
| "transformer_blocks.1.attn.to_v": 836, |
| "transformer_blocks.1.ff.net.0.proj": 1642, |
| "transformer_blocks.1.ff.net.2": 2450, |
| "transformer_blocks.10.attn.to_k": 273, |
| "transformer_blocks.10.attn.to_out.0": 388, |
| "transformer_blocks.10.attn.to_q": 265, |
| "transformer_blocks.10.attn.to_v": 448, |
| "transformer_blocks.10.ff.net.0.proj": 335, |
| "transformer_blocks.10.ff.net.2": 1478, |
| "transformer_blocks.11.attn.to_k": 256, |
| "transformer_blocks.11.attn.to_out.0": 410, |
| "transformer_blocks.11.attn.to_q": 264, |
| "transformer_blocks.11.attn.to_v": 445, |
| "transformer_blocks.11.ff.net.0.proj": 361, |
| "transformer_blocks.11.ff.net.2": 456, |
| "transformer_blocks.12.attn.to_k": 145, |
| "transformer_blocks.12.attn.to_out.0": 201, |
| "transformer_blocks.12.attn.to_q": 155, |
| "transformer_blocks.12.attn.to_v": 300, |
| "transformer_blocks.12.ff.net.0.proj": 210, |
| "transformer_blocks.12.ff.net.2": 1082, |
| "transformer_blocks.13.attn.to_k": 559, |
| "transformer_blocks.13.attn.to_out.0": 696, |
| "transformer_blocks.13.attn.to_q": 565, |
| "transformer_blocks.13.attn.to_v": 790, |
| "transformer_blocks.13.ff.net.0.proj": 734, |
| "transformer_blocks.13.ff.net.2": 2, |
| "transformer_blocks.14.attn.to_k": 238, |
| "transformer_blocks.14.attn.to_out.0": 357, |
| "transformer_blocks.14.attn.to_q": 258, |
| "transformer_blocks.14.attn.to_v": 438, |
| "transformer_blocks.14.ff.net.0.proj": 315, |
| "transformer_blocks.14.ff.net.2": 1566, |
| "transformer_blocks.15.attn.to_k": 248, |
| "transformer_blocks.15.attn.to_out.0": 394, |
| "transformer_blocks.15.attn.to_q": 261, |
| "transformer_blocks.15.attn.to_v": 431, |
| "transformer_blocks.15.ff.net.0.proj": 315, |
| "transformer_blocks.15.ff.net.2": 1539, |
| "transformer_blocks.16.attn.to_k": 451, |
| "transformer_blocks.16.attn.to_out.0": 711, |
| "transformer_blocks.16.attn.to_q": 460, |
| "transformer_blocks.16.attn.to_v": 694, |
| "transformer_blocks.16.ff.net.0.proj": 652, |
| "transformer_blocks.16.ff.net.2": 2287, |
| "transformer_blocks.17.attn.to_k": 242, |
| "transformer_blocks.17.attn.to_out.0": 413, |
| "transformer_blocks.17.attn.to_q": 261, |
| "transformer_blocks.17.attn.to_v": 443, |
| "transformer_blocks.17.ff.net.0.proj": 334, |
| "transformer_blocks.17.ff.net.2": 1489, |
| "transformer_blocks.18.attn.to_k": 210, |
| "transformer_blocks.18.attn.to_out.0": 440, |
| "transformer_blocks.18.attn.to_q": 238, |
| "transformer_blocks.18.attn.to_v": 433, |
| "transformer_blocks.18.ff.net.0.proj": 390, |
| "transformer_blocks.18.ff.net.2": 1417, |
| "transformer_blocks.2.attn.to_k": 1551, |
| "transformer_blocks.2.attn.to_out.0": 2092, |
| "transformer_blocks.2.attn.to_q": 1587, |
| "transformer_blocks.2.attn.to_v": 1615, |
| "transformer_blocks.2.ff.net.0.proj": 3137, |
| "transformer_blocks.20.attn.to_k": 814, |
| "transformer_blocks.20.attn.to_out.0": 1209, |
| "transformer_blocks.20.attn.to_q": 831, |
| "transformer_blocks.20.attn.to_v": 1093, |
| "transformer_blocks.20.ff.net.0.proj": 1300, |
| "transformer_blocks.20.ff.net.2": 3061, |
| "transformer_blocks.23.attn.to_k": 70, |
| "transformer_blocks.23.attn.to_out.0": 267, |
| "transformer_blocks.23.attn.to_q": 102, |
| "transformer_blocks.23.attn.to_v": 241, |
| "transformer_blocks.23.ff.net.0.proj": 232, |
| "transformer_blocks.23.ff.net.2": 1121, |
| "transformer_blocks.4.attn.to_k": 182, |
| "transformer_blocks.4.attn.to_out.0": 255, |
| "transformer_blocks.4.attn.to_q": 170, |
| "transformer_blocks.4.attn.to_v": 280, |
| "transformer_blocks.4.ff.net.0.proj": 333, |
| "transformer_blocks.4.ff.net.2": 815, |
| "transformer_blocks.49.ff.net.2": 2932, |
| "transformer_blocks.7.attn.to_k": 186, |
| "transformer_blocks.7.attn.to_out.0": 315, |
| "transformer_blocks.7.attn.to_q": 197, |
| "transformer_blocks.7.attn.to_v": 330, |
| "transformer_blocks.7.ff.net.0.proj": 265, |
| "transformer_blocks.7.ff.net.2": 843, |
| "transformer_blocks.8.attn.to_k": 129, |
| "transformer_blocks.8.attn.to_out.0": 224, |
| "transformer_blocks.8.attn.to_q": 136, |
| "transformer_blocks.8.attn.to_v": 255, |
| "transformer_blocks.8.ff.net.0.proj": 209, |
| "transformer_blocks.8.ff.net.2": 1120, |
| "transformer_blocks.9.attn.to_k": 132, |
| "transformer_blocks.9.attn.to_out.0": 270, |
| "transformer_blocks.9.attn.to_q": 146, |
| "transformer_blocks.9.attn.to_v": 260, |
| "transformer_blocks.9.ff.net.0.proj": 185, |
| "transformer_blocks.9.ff.net.2": 1121 |
| }, |
| "dropped_blocks": [ |
| 6, |
| 21, |
| 22 |
| ], |
| "auto_map": { |
| "AutoModel": "hyperflow_h3_flash_transformer.HyperFlowH3FlashTransformer3DModel" |
| }, |
| "quantization": { |
| "method": "int8_convrot", |
| "group": 256, |
| "linears": [ |
| "transformer_blocks.0.attn.to_k", |
| "transformer_blocks.0.attn.to_out.0", |
| "transformer_blocks.0.attn.to_q", |
| "transformer_blocks.0.attn.to_v", |
| "transformer_blocks.0.ff.net.0.proj", |
| "transformer_blocks.0.ff.net.2", |
| "transformer_blocks.19.attn.to_k", |
| "transformer_blocks.19.attn.to_out.0", |
| "transformer_blocks.19.attn.to_q", |
| "transformer_blocks.19.attn.to_v", |
| "transformer_blocks.19.ff.net.0.proj", |
| "transformer_blocks.19.ff.net.2", |
| "transformer_blocks.2.ff.net.2", |
| "transformer_blocks.24.attn.to_k", |
| "transformer_blocks.24.attn.to_out.0", |
| "transformer_blocks.24.attn.to_q", |
| "transformer_blocks.24.attn.to_v", |
| "transformer_blocks.24.ff.net.0.proj", |
| "transformer_blocks.24.ff.net.2", |
| "transformer_blocks.25.attn.to_k", |
| "transformer_blocks.25.attn.to_out.0", |
| "transformer_blocks.25.attn.to_q", |
| "transformer_blocks.25.attn.to_v", |
| "transformer_blocks.25.ff.net.0.proj", |
| "transformer_blocks.25.ff.net.2", |
| "transformer_blocks.26.attn.to_k", |
| "transformer_blocks.26.attn.to_out.0", |
| "transformer_blocks.26.attn.to_q", |
| "transformer_blocks.26.attn.to_v", |
| "transformer_blocks.26.ff.net.0.proj", |
| "transformer_blocks.26.ff.net.2", |
| "transformer_blocks.27.attn.to_k", |
| "transformer_blocks.27.attn.to_out.0", |
| "transformer_blocks.27.attn.to_q", |
| "transformer_blocks.27.attn.to_v", |
| "transformer_blocks.27.ff.net.0.proj", |
| "transformer_blocks.27.ff.net.2", |
| "transformer_blocks.28.attn.to_k", |
| "transformer_blocks.28.attn.to_out.0", |
| "transformer_blocks.28.attn.to_q", |
| "transformer_blocks.28.attn.to_v", |
| "transformer_blocks.28.ff.net.0.proj", |
| "transformer_blocks.28.ff.net.2", |
| "transformer_blocks.29.attn.to_k", |
| "transformer_blocks.29.attn.to_out.0", |
| "transformer_blocks.29.attn.to_q", |
| "transformer_blocks.29.attn.to_v", |
| "transformer_blocks.29.ff.net.0.proj", |
| "transformer_blocks.29.ff.net.2", |
| "transformer_blocks.3.attn.to_k", |
| "transformer_blocks.3.attn.to_out.0", |
| "transformer_blocks.3.attn.to_q", |
| "transformer_blocks.3.attn.to_v", |
| "transformer_blocks.3.ff.net.0.proj", |
| "transformer_blocks.3.ff.net.2", |
| "transformer_blocks.30.attn.to_k", |
| "transformer_blocks.30.attn.to_out.0", |
| "transformer_blocks.30.attn.to_q", |
| "transformer_blocks.30.attn.to_v", |
| "transformer_blocks.30.ff.net.0.proj", |
| "transformer_blocks.30.ff.net.2", |
| "transformer_blocks.31.attn.to_k", |
| "transformer_blocks.31.attn.to_out.0", |
| "transformer_blocks.31.attn.to_q", |
| "transformer_blocks.31.attn.to_v", |
| "transformer_blocks.31.ff.net.0.proj", |
| "transformer_blocks.31.ff.net.2", |
| "transformer_blocks.32.attn.to_k", |
| "transformer_blocks.32.attn.to_out.0", |
| "transformer_blocks.32.attn.to_q", |
| "transformer_blocks.32.attn.to_v", |
| "transformer_blocks.32.ff.net.0.proj", |
| "transformer_blocks.32.ff.net.2", |
| "transformer_blocks.33.attn.to_k", |
| "transformer_blocks.33.attn.to_out.0", |
| "transformer_blocks.33.attn.to_q", |
| "transformer_blocks.33.attn.to_v", |
| "transformer_blocks.33.ff.net.0.proj", |
| "transformer_blocks.33.ff.net.2", |
| "transformer_blocks.34.attn.to_k", |
| "transformer_blocks.34.attn.to_out.0", |
| "transformer_blocks.34.attn.to_q", |
| "transformer_blocks.34.attn.to_v", |
| "transformer_blocks.34.ff.net.0.proj", |
| "transformer_blocks.34.ff.net.2", |
| "transformer_blocks.35.attn.to_k", |
| "transformer_blocks.35.attn.to_out.0", |
| "transformer_blocks.35.attn.to_q", |
| "transformer_blocks.35.attn.to_v", |
| "transformer_blocks.35.ff.net.0.proj", |
| "transformer_blocks.35.ff.net.2", |
| "transformer_blocks.36.attn.to_k", |
| "transformer_blocks.36.attn.to_out.0", |
| "transformer_blocks.36.attn.to_q", |
| "transformer_blocks.36.attn.to_v", |
| "transformer_blocks.36.ff.net.0.proj", |
| "transformer_blocks.36.ff.net.2", |
| "transformer_blocks.37.attn.to_k", |
| "transformer_blocks.37.attn.to_out.0", |
| "transformer_blocks.37.attn.to_q", |
| "transformer_blocks.37.attn.to_v", |
| "transformer_blocks.37.ff.net.0.proj", |
| "transformer_blocks.37.ff.net.2", |
| "transformer_blocks.38.attn.to_k", |
| "transformer_blocks.38.attn.to_out.0", |
| "transformer_blocks.38.attn.to_q", |
| "transformer_blocks.38.attn.to_v", |
| "transformer_blocks.38.ff.net.0.proj", |
| "transformer_blocks.38.ff.net.2", |
| "transformer_blocks.39.attn.to_k", |
| "transformer_blocks.39.attn.to_out.0", |
| "transformer_blocks.39.attn.to_q", |
| "transformer_blocks.39.attn.to_v", |
| "transformer_blocks.39.ff.net.0.proj", |
| "transformer_blocks.39.ff.net.2", |
| "transformer_blocks.40.attn.to_k", |
| "transformer_blocks.40.attn.to_out.0", |
| "transformer_blocks.40.attn.to_q", |
| "transformer_blocks.40.attn.to_v", |
| "transformer_blocks.40.ff.net.0.proj", |
| "transformer_blocks.40.ff.net.2", |
| "transformer_blocks.41.attn.to_k", |
| "transformer_blocks.41.attn.to_out.0", |
| "transformer_blocks.41.attn.to_q", |
| "transformer_blocks.41.attn.to_v", |
| "transformer_blocks.41.ff.net.0.proj", |
| "transformer_blocks.41.ff.net.2", |
| "transformer_blocks.42.attn.to_k", |
| "transformer_blocks.42.attn.to_out.0", |
| "transformer_blocks.42.attn.to_q", |
| "transformer_blocks.42.attn.to_v", |
| "transformer_blocks.42.ff.net.0.proj", |
| "transformer_blocks.42.ff.net.2", |
| "transformer_blocks.43.attn.to_k", |
| "transformer_blocks.43.attn.to_out.0", |
| "transformer_blocks.43.attn.to_q", |
| "transformer_blocks.43.attn.to_v", |
| "transformer_blocks.43.ff.net.0.proj", |
| "transformer_blocks.43.ff.net.2", |
| "transformer_blocks.44.attn.to_k", |
| "transformer_blocks.44.attn.to_out.0", |
| "transformer_blocks.44.attn.to_q", |
| "transformer_blocks.44.attn.to_v", |
| "transformer_blocks.44.ff.net.0.proj", |
| "transformer_blocks.44.ff.net.2", |
| "transformer_blocks.45.attn.to_k", |
| "transformer_blocks.45.attn.to_out.0", |
| "transformer_blocks.45.attn.to_q", |
| "transformer_blocks.45.attn.to_v", |
| "transformer_blocks.45.ff.net.0.proj", |
| "transformer_blocks.45.ff.net.2", |
| "transformer_blocks.46.attn.to_k", |
| "transformer_blocks.46.attn.to_out.0", |
| "transformer_blocks.46.attn.to_q", |
| "transformer_blocks.46.attn.to_v", |
| "transformer_blocks.46.ff.net.0.proj", |
| "transformer_blocks.46.ff.net.2", |
| "transformer_blocks.47.attn.to_k", |
| "transformer_blocks.47.attn.to_out.0", |
| "transformer_blocks.47.attn.to_q", |
| "transformer_blocks.47.attn.to_v", |
| "transformer_blocks.47.ff.net.0.proj", |
| "transformer_blocks.47.ff.net.2", |
| "transformer_blocks.48.attn.to_k", |
| "transformer_blocks.48.attn.to_out.0", |
| "transformer_blocks.48.attn.to_q", |
| "transformer_blocks.48.attn.to_v", |
| "transformer_blocks.48.ff.net.0.proj", |
| "transformer_blocks.48.ff.net.2", |
| "transformer_blocks.49.attn.to_k", |
| "transformer_blocks.49.attn.to_out.0", |
| "transformer_blocks.49.attn.to_q", |
| "transformer_blocks.49.attn.to_v", |
| "transformer_blocks.49.ff.net.0.proj", |
| "transformer_blocks.5.attn.to_k", |
| "transformer_blocks.5.attn.to_out.0", |
| "transformer_blocks.5.attn.to_q", |
| "transformer_blocks.5.attn.to_v", |
| "transformer_blocks.5.ff.net.0.proj", |
| "transformer_blocks.5.ff.net.2" |
| ] |
| } |
| } |
|
|
