jennyxue's picture
bf16 build of HyperFlow H3 Flash
f9b8992 verified
Raw History Blame Contribute Delete
5.23 kB
{
"_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": 704,
"transformer_blocks.1.attn.to_out.0": 1792,
"transformer_blocks.1.attn.to_q": 768,
"transformer_blocks.1.attn.to_v": 896,
"transformer_blocks.1.ff.net.0.proj": 1664,
"transformer_blocks.1.ff.net.2": 2496,
"transformer_blocks.10.attn.to_k": 320,
"transformer_blocks.10.attn.to_out.0": 448,
"transformer_blocks.10.attn.to_q": 320,
"transformer_blocks.10.attn.to_v": 448,
"transformer_blocks.10.ff.net.0.proj": 384,
"transformer_blocks.10.ff.net.2": 1536,
"transformer_blocks.11.attn.to_k": 256,
"transformer_blocks.11.attn.to_out.0": 448,
"transformer_blocks.11.attn.to_q": 320,
"transformer_blocks.11.attn.to_v": 448,
"transformer_blocks.11.ff.net.0.proj": 384,
"transformer_blocks.11.ff.net.2": 512,
"transformer_blocks.12.attn.to_k": 192,
"transformer_blocks.12.attn.to_out.0": 256,
"transformer_blocks.12.attn.to_q": 192,
"transformer_blocks.12.attn.to_v": 320,
"transformer_blocks.12.ff.net.0.proj": 256,
"transformer_blocks.12.ff.net.2": 1088,
"transformer_blocks.13.attn.to_k": 576,
"transformer_blocks.13.attn.to_out.0": 704,
"transformer_blocks.13.attn.to_q": 576,
"transformer_blocks.13.attn.to_v": 832,
"transformer_blocks.13.ff.net.0.proj": 768,
"transformer_blocks.13.ff.net.2": 64,
"transformer_blocks.14.attn.to_k": 256,
"transformer_blocks.14.attn.to_out.0": 384,
"transformer_blocks.14.attn.to_q": 320,
"transformer_blocks.14.attn.to_v": 448,
"transformer_blocks.14.ff.net.0.proj": 320,
"transformer_blocks.14.ff.net.2": 1600,
"transformer_blocks.15.attn.to_k": 256,
"transformer_blocks.15.attn.to_out.0": 448,
"transformer_blocks.15.attn.to_q": 320,
"transformer_blocks.15.attn.to_v": 448,
"transformer_blocks.15.ff.net.0.proj": 320,
"transformer_blocks.15.ff.net.2": 1600,
"transformer_blocks.16.attn.to_k": 512,
"transformer_blocks.16.attn.to_out.0": 768,
"transformer_blocks.16.attn.to_q": 512,
"transformer_blocks.16.attn.to_v": 704,
"transformer_blocks.16.ff.net.0.proj": 704,
"transformer_blocks.16.ff.net.2": 2304,
"transformer_blocks.17.attn.to_k": 256,
"transformer_blocks.17.attn.to_out.0": 448,
"transformer_blocks.17.attn.to_q": 320,
"transformer_blocks.17.attn.to_v": 448,
"transformer_blocks.17.ff.net.0.proj": 384,
"transformer_blocks.17.ff.net.2": 1536,
"transformer_blocks.18.attn.to_k": 256,
"transformer_blocks.18.attn.to_out.0": 448,
"transformer_blocks.18.attn.to_q": 256,
"transformer_blocks.18.attn.to_v": 448,
"transformer_blocks.18.ff.net.0.proj": 448,
"transformer_blocks.18.ff.net.2": 1472,
"transformer_blocks.2.attn.to_k": 1600,
"transformer_blocks.2.attn.to_out.0": 2112,
"transformer_blocks.2.attn.to_q": 1600,
"transformer_blocks.2.attn.to_v": 1664,
"transformer_blocks.2.ff.net.0.proj": 3200,
"transformer_blocks.20.attn.to_k": 832,
"transformer_blocks.20.attn.to_out.0": 1216,
"transformer_blocks.20.attn.to_q": 832,
"transformer_blocks.20.attn.to_v": 1152,
"transformer_blocks.20.ff.net.0.proj": 1344,
"transformer_blocks.20.ff.net.2": 3072,
"transformer_blocks.23.attn.to_k": 128,
"transformer_blocks.23.attn.to_out.0": 320,
"transformer_blocks.23.attn.to_q": 128,
"transformer_blocks.23.attn.to_v": 256,
"transformer_blocks.23.ff.net.0.proj": 256,
"transformer_blocks.23.ff.net.2": 1152,
"transformer_blocks.4.attn.to_k": 192,
"transformer_blocks.4.attn.to_out.0": 256,
"transformer_blocks.4.attn.to_q": 192,
"transformer_blocks.4.attn.to_v": 320,
"transformer_blocks.4.ff.net.0.proj": 384,
"transformer_blocks.4.ff.net.2": 832,
"transformer_blocks.49.ff.net.2": 2944,
"transformer_blocks.7.attn.to_k": 192,
"transformer_blocks.7.attn.to_out.0": 320,
"transformer_blocks.7.attn.to_q": 256,
"transformer_blocks.7.attn.to_v": 384,
"transformer_blocks.7.ff.net.0.proj": 320,
"transformer_blocks.7.ff.net.2": 896,
"transformer_blocks.8.attn.to_k": 192,
"transformer_blocks.8.attn.to_out.0": 256,
"transformer_blocks.8.attn.to_q": 192,
"transformer_blocks.8.attn.to_v": 256,
"transformer_blocks.8.ff.net.0.proj": 256,
"transformer_blocks.8.ff.net.2": 1152,
"transformer_blocks.9.attn.to_k": 192,
"transformer_blocks.9.attn.to_out.0": 320,
"transformer_blocks.9.attn.to_q": 192,
"transformer_blocks.9.attn.to_v": 320,
"transformer_blocks.9.ff.net.0.proj": 192,
"transformer_blocks.9.ff.net.2": 1152
},
"dropped_blocks": [
6,
21,
22
],
"source": {
"repo": "VideoRebirthAI/hyperflow-h3-flash",
"revision": "c8f74e481b9f7743cd91d2d3c4bf108eeb723748",
"folder": "fl2va_int8_convrot",
"quantization": {
"method": "int8_convrot",
"group": 256
}
}
}