suvadityamuk's picture
suvadityamuk HF Staff
Convert microsoft/TRELLIS.2-4B with diffusers-3d 0.1.0.dev0
20af78a verified
Raw History Blame Contribute Delete
513 Bytes
{
"_class_name": "Trellis2SparseStructureFlowModel",
"_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",
"qk_rms_norm": true,
"qk_rms_norm_cross": true,
"resolution": 16,
"rope_freq": [
1.0,
10000.0
],
"share_mod": true,
"use_checkpoint": false
}