| { |
| "_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": 990, |
| "transformer_blocks.1.attn.to_out.0": 1872, |
| "transformer_blocks.1.attn.to_q": 1102, |
| "transformer_blocks.1.attn.to_v": 1200, |
| "transformer_blocks.1.ff.net.0.proj": 2516, |
| "transformer_blocks.1.ff.net.2": 3120, |
| "transformer_blocks.10.attn.to_k": 275, |
| "transformer_blocks.10.attn.to_out.0": 362, |
| "transformer_blocks.10.attn.to_q": 269, |
| "transformer_blocks.10.attn.to_v": 471, |
| "transformer_blocks.10.ff.net.0.proj": 359, |
| "transformer_blocks.10.ff.net.2": 1436, |
| "transformer_blocks.11.attn.to_k": 420, |
| "transformer_blocks.11.attn.to_out.0": 572, |
| "transformer_blocks.11.attn.to_q": 426, |
| "transformer_blocks.11.attn.to_v": 673, |
| "transformer_blocks.11.ff.net.0.proj": 604, |
| "transformer_blocks.11.ff.net.2": 557, |
| "transformer_blocks.12.attn.to_k": 171, |
| "transformer_blocks.12.attn.to_out.0": 210, |
| "transformer_blocks.12.attn.to_q": 181, |
| "transformer_blocks.12.attn.to_v": 342, |
| "transformer_blocks.12.ff.net.0.proj": 247, |
| "transformer_blocks.12.ff.net.2": 953, |
| "transformer_blocks.13.attn.to_k": 379, |
| "transformer_blocks.13.attn.to_out.0": 474, |
| "transformer_blocks.13.attn.to_q": 381, |
| "transformer_blocks.13.attn.to_v": 582, |
| "transformer_blocks.13.ff.net.0.proj": 462, |
| "transformer_blocks.13.ff.net.2": 1, |
| "transformer_blocks.14.attn.to_k": 286, |
| "transformer_blocks.14.attn.to_out.0": 376, |
| "transformer_blocks.14.attn.to_q": 305, |
| "transformer_blocks.14.attn.to_v": 505, |
| "transformer_blocks.14.ff.net.0.proj": 374, |
| "transformer_blocks.14.ff.net.2": 1348, |
| "transformer_blocks.15.attn.to_k": 155, |
| "transformer_blocks.15.attn.to_out.0": 261, |
| "transformer_blocks.15.attn.to_q": 167, |
| "transformer_blocks.15.attn.to_v": 311, |
| "transformer_blocks.15.ff.net.0.proj": 201, |
| "transformer_blocks.15.ff.net.2": 1038, |
| "transformer_blocks.16.attn.to_k": 239, |
| "transformer_blocks.16.attn.to_out.0": 400, |
| "transformer_blocks.16.attn.to_q": 246, |
| "transformer_blocks.16.attn.to_v": 440, |
| "transformer_blocks.16.ff.net.0.proj": 335, |
| "transformer_blocks.16.ff.net.2": 1510, |
| "transformer_blocks.17.attn.to_k": 951, |
| "transformer_blocks.17.attn.to_out.0": 1116, |
| "transformer_blocks.17.attn.to_q": 952, |
| "transformer_blocks.17.attn.to_v": 1216, |
| "transformer_blocks.17.ff.net.0.proj": 1325, |
| "transformer_blocks.17.ff.net.2": 2836, |
| "transformer_blocks.18.attn.to_k": 121, |
| "transformer_blocks.18.attn.to_out.0": 279, |
| "transformer_blocks.18.attn.to_q": 140, |
| "transformer_blocks.18.attn.to_v": 287, |
| "transformer_blocks.18.ff.net.0.proj": 234, |
| "transformer_blocks.18.ff.net.2": 1024, |
| "transformer_blocks.20.attn.to_k": 433, |
| "transformer_blocks.20.attn.to_out.0": 725, |
| "transformer_blocks.20.attn.to_q": 445, |
| "transformer_blocks.20.attn.to_v": 656, |
| "transformer_blocks.20.ff.net.0.proj": 698, |
| "transformer_blocks.20.ff.net.2": 2243, |
| "transformer_blocks.21.attn.to_k": 389, |
| "transformer_blocks.21.attn.to_out.0": 704, |
| "transformer_blocks.21.attn.to_q": 434, |
| "transformer_blocks.21.attn.to_v": 669, |
| "transformer_blocks.21.ff.net.0.proj": 711, |
| "transformer_blocks.21.ff.net.2": 2261, |
| "transformer_blocks.22.attn.to_k": 170, |
| "transformer_blocks.22.attn.to_out.0": 459, |
| "transformer_blocks.22.attn.to_q": 199, |
| "transformer_blocks.22.attn.to_v": 385, |
| "transformer_blocks.22.ff.net.0.proj": 378, |
| "transformer_blocks.22.ff.net.2": 1507, |
| "transformer_blocks.23.attn.to_k": 78, |
| "transformer_blocks.23.attn.to_out.0": 261, |
| "transformer_blocks.23.attn.to_q": 103, |
| "transformer_blocks.23.attn.to_v": 227, |
| "transformer_blocks.23.ff.net.0.proj": 221, |
| "transformer_blocks.23.ff.net.2": 1147, |
| "transformer_blocks.4.attn.to_k": 284, |
| "transformer_blocks.4.attn.to_out.0": 398, |
| "transformer_blocks.4.attn.to_q": 265, |
| "transformer_blocks.4.attn.to_v": 403, |
| "transformer_blocks.4.ff.net.0.proj": 546, |
| "transformer_blocks.4.ff.net.2": 1130, |
| "transformer_blocks.5.attn.to_k": 224, |
| "transformer_blocks.5.attn.to_out.0": 424, |
| "transformer_blocks.5.attn.to_q": 224, |
| "transformer_blocks.5.attn.to_v": 413, |
| "transformer_blocks.5.ff.net.0.proj": 991, |
| "transformer_blocks.5.ff.net.2": 2357, |
| "transformer_blocks.6.attn.to_k": 439, |
| "transformer_blocks.6.attn.to_out.0": 673, |
| "transformer_blocks.6.attn.to_q": 428, |
| "transformer_blocks.6.attn.to_v": 607, |
| "transformer_blocks.6.ff.net.0.proj": 836, |
| "transformer_blocks.6.ff.net.2": 2076, |
| "transformer_blocks.7.attn.to_k": 621, |
| "transformer_blocks.7.attn.to_out.0": 815, |
| "transformer_blocks.7.attn.to_q": 626, |
| "transformer_blocks.7.attn.to_v": 864, |
| "transformer_blocks.7.ff.net.0.proj": 986, |
| "transformer_blocks.7.ff.net.2": 1322, |
| "transformer_blocks.8.attn.to_k": 483, |
| "transformer_blocks.8.attn.to_out.0": 609, |
| "transformer_blocks.8.attn.to_q": 479, |
| "transformer_blocks.8.attn.to_v": 704, |
| "transformer_blocks.8.ff.net.0.proj": 802, |
| "transformer_blocks.8.ff.net.2": 2054 |
| }, |
| "dropped_blocks": [ |
| 9 |
| ], |
| "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.attn.to_k", |
| "transformer_blocks.2.attn.to_out.0", |
| "transformer_blocks.2.attn.to_q", |
| "transformer_blocks.2.attn.to_v", |
| "transformer_blocks.2.ff.net.0.proj", |
| "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.49.ff.net.2" |
| ] |
| } |
| } |
|
|
