timasher/pitbandit-example

1{2  "_class_name": "UNet2DConditionModel",3  "_diffusers_version": "0.15.0.dev0",4  "act_fn": "silu",5  "attention_head_dim": 8,6  "block_out_channels": [7    320,8    640,9    1280,10    128011  ],12  "center_input_sample": false,13  "class_embed_type": null,14  "conv_in_kernel": 3,15  "conv_out_kernel": 3,16  "cross_attention_dim": 768,17  "down_block_types": [18    "CrossAttnDownBlock2D",19    "CrossAttnDownBlock2D",20    "CrossAttnDownBlock2D",21    "DownBlock2D"22  ],23  "downsample_padding": 1,24  "dual_cross_attention": false,25  "flip_sin_to_cos": true,26  "freq_shift": 0,27  "in_channels": 4,28  "layers_per_block": 2,29  "mid_block_scale_factor": 1,30  "mid_block_type": "UNetMidBlock2DCrossAttn",31  "norm_eps": 1e-05,32  "norm_num_groups": 32,33  "num_class_embeds": null,34  "only_cross_attention": false,35  "out_channels": 4,36  "projection_class_embeddings_input_dim": null,37  "resnet_time_scale_shift": "default",38  "sample_size": 64,39  "time_cond_proj_dim": null,40  "time_embedding_type": "positional",41  "timestep_post_act": null,42  "up_block_types": [43    "UpBlock2D",44    "CrossAttnUpBlock2D",45    "CrossAttnUpBlock2D",46    "CrossAttnUpBlock2D"47  ],48  "upcast_attention": false,49  "use_linear_projection": false50}51