File size: 540 Bytes
9d37047 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | {
"_class_name": "Pixal3DSparseStructureFlowModel",
"_diffusers_version": "0.41.0.dev0",
"cond_channels": 1024,
"dtype": "bfloat16",
"in_channels": 8,
"initialization": "scaled",
"mlp_ratio": 5.3334,
"model_channels": 1536,
"num_blocks": 30,
"num_head_channels": 64,
"num_heads": 12,
"out_channels": 8,
"pe_mode": "rope",
"proj_in_channels": 1024,
"qk_rms_norm": true,
"qk_rms_norm_cross": true,
"resolution": 16,
"rope_freq": [
1.0,
10000.0
],
"share_mod": true,
"use_checkpoint": false
}
|