sayakpaul's picture
sayakpaul HF Staff
Upload DiTPipeline
9cb097e verified
{
"_class_name": "DiTTransformer2DModel",
"_diffusers_version": "0.29.0.dev0",
"activation_fn": "gelu-approximate",
"attention_bias": true,
"attention_head_dim": 8,
"dropout": 0.0,
"in_channels": 4,
"norm_elementwise_affine": false,
"norm_eps": 1e-05,
"norm_num_groups": 32,
"norm_type": "ada_norm_zero",
"num_attention_heads": 2,
"num_embeds_ada_norm": 1000,
"num_layers": 2,
"out_channels": 8,
"patch_size": 4,
"sample_size": 16,
"upcast_attention": false
}