Struct ZConfig 

pub struct ZConfig {

    pub dim: usize,
    pub n_layers: usize,
    pub n_refiner: usize,
    pub n_heads: usize,
    pub n_kv_heads: usize,
    pub head_dim: usize,
    pub ffn_dim: usize,
    pub cap_feat_dim: usize,
    pub t_embed_dim: usize,
    pub t_hidden: usize,
    pub patch: usize,
    pub in_channels: usize,
    pub norm_eps: f32,
    pub final_eps: f32,
    pub rope_theta: f64,
    pub axes_dims: [usize; 3],
    pub t_scale: f32,
}

dit.config_json / transformer config.json fields the runtime uses.

Fields

dim: usize

3840

n_layers: usize

30 main layers

n_refiner: usize

2 noise-refiner and 2 context-refiner blocks

n_heads: usize

30 heads, 30 kv heads, head_dim 128

n_kv_heads: usize

head_dim: usize

ffn_dim: usize

int(dim / 3 · 8) = 10240

cap_feat_dim: usize

2560 (Qwen3-4B hidden)

t_embed_dim: usize

256 (ADALN_EMBED_DIM) and the t-embedder MLP width 1024

t_hidden: usize

patch: usize

patch 2, latent channels 16 → patch vector 64

in_channels: usize

norm_eps: f32

1e-5 (all RMSNorms), 1e-6 (final LayerNorm)

final_eps: f32

rope_theta: f64

256.0 and [32, 48, 48]

axes_dims: [usize; 3]

t_scale: f32

1000.0

Implementations

impl ZConfig

pub fn from_json(v: &Value) -> Result<Self, String>

Parse the diffusers transformer config.json (the same JSON is stored as dit.config_json in the container).

pub fn geom(&self) -> ZGeom

pub fn n_mod_blocks(&self) -> usize

Blocks carrying adaLN: the noise refiner, then the main layers.

Trait Implementations

impl Clone for ZConfig

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ZConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.