slat_flow_model/config.json

{
  "_class_name": "TrellisSLatFlowModel",
  "_diffusers_version": "0.41.0.dev0",
  "cond_channels": 1024,
  "in_channels": 8,
  "io_block_channels": [
    128
  ],
  "mlp_ratio": 4,
  "model_channels": 1024,
  "num_blocks": 24,
  "num_head_channels": 64,
  "num_heads": 16,
  "num_io_res_blocks": 2,
  "out_channels": 8,
  "patch_size": 2,
  "pe_mode": "ape",
  "qk_rms_norm": true,
  "qk_rms_norm_cross": false,
  "resolution": 64,
  "share_mod": false,
  "use_checkpoint": false,
  "use_fp16": true,
  "use_skip_connection": true
}