{ "_class_name": "FlaxControlNetModel", "_diffusers_version": "0.16.0.dev0", "attention_head_dim": 8, "block_out_channels": [ 320, 640, 1280, 1280 ], "conditioning_embedding_out_channels": [ 16, 32, 96, 256 ], "controlnet_conditioning_channel_order": "rgb", "cross_attention_dim": 768, "down_block_types": [ "CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "DownBlock2D" ], "dropout": 0.0, "flip_sin_to_cos": true, "freq_shift": 0, "in_channels": 4, "layers_per_block": 2, "only_cross_attention": false, "sample_size": 32, "use_linear_projection": false }