Allegro/transformer/config.json
YuxuanCai e3d6332920
Add model_index.json and update modeling configs (#2)
- add model_index.json; update configs (43d06687e1b23afacab5980e452638293b3722b0)
- update (7defb981dbb797a900f485c84f483d6d21e9520a)


Co-authored-by: Aryan V S <a-r-r-o-w@users.noreply.huggingface.co>
2024-10-29 03:50:46 +00:00

31 lines
707 B
JSON

{
"_class_name": "AllegroTransformer3DModel",
"_diffusers_version": "0.31.0.dev0",
"activation_fn": "gelu-approximate",
"attention_bias": true,
"attention_head_dim": 96,
"caption_channels": 4096,
"cross_attention_dim": 2304,
"dropout": 0.0,
"in_channels": 4,
"interpolation_scale_h": 2.0,
"interpolation_scale_t": 2.2,
"interpolation_scale_w": 2.0,
"norm_elementwise_affine": false,
"norm_eps": 1e-06,
"norm_type": "ada_norm_single",
"num_attention_heads": 24,
"num_layers": 32,
"out_channels": 4,
"patch_size": 2,
"patch_size_t": 1,
"sample_frames": 22,
"sample_height": 90,
"sample_size": [
90,
160
],
"sample_size_t": 22,
"sample_width": 160
}