YiYiXu's picture
Update transformer/config.json
7e7b7a1 verified
raw
history blame contribute delete
495 Bytes
{
"_class_name": "HunyuanImageTransformer2DModel",
"_diffusers_version": "0.36.0.dev0",
"attention_head_dim": 128,
"guidance_embeds": false,
"in_channels": 64,
"mlp_ratio": 4.0,
"num_attention_heads": 28,
"num_layers": 20,
"num_refiner_layers": 2,
"num_single_layers": 40,
"out_channels": 64,
"patch_size": [
1,
1
],
"qk_norm": "rms_norm",
"rope_axes_dim": [
64,
64
],
"rope_theta": 256.0,
"text_embed_2_dim": 1472,
"text_embed_dim": 3584
}