{ | |
"_class_name": "FlaxUNet2DConditionModel", | |
"_diffusers_version": "0.18.0.dev0", | |
"_name_or_path": "./unet", | |
"addition_embed_type": "text_time", | |
"addition_embed_type_num_heads": 64, | |
"addition_time_embed_dim": 256, | |
"attention_head_dim": [ | |
5, | |
10, | |
20 | |
], | |
"block_out_channels": [ | |
320, | |
640, | |
1280 | |
], | |
"cross_attention_dim": 2048, | |
"down_block_types": [ | |
"DownBlock2D", | |
"CrossAttnDownBlock2D", | |
"CrossAttnDownBlock2D" | |
], | |
"dropout": 0.0, | |
"flip_sin_to_cos": true, | |
"freq_shift": 0, | |
"in_channels": 4, | |
"layers_per_block": 2, | |
"num_attention_heads": null, | |
"only_cross_attention": false, | |
"out_channels": 4, | |
"projection_class_embeddings_input_dim": 2816, | |
"sample_size": 128, | |
"transformer_layers_per_block": [ | |
1, | |
2, | |
10 | |
], | |
"up_block_types": [ | |
"CrossAttnUpBlock2D", | |
"CrossAttnUpBlock2D", | |
"UpBlock2D" | |
], | |
"use_linear_projection": true, | |
"use_memory_efficient_attention": false | |
} | |