UmitDataTeam/fine-diffusion

1{2    "_class_name": "UNet2DConditionModel",3    "_diffusers_version": "0.6.0",4    "act_fn": "silu",5    "attention_head_dim": 8,6    "block_out_channels": [7      320,8      640,9      128010    ],11    "center_input_sample": false,12    "cross_attention_dim": 768,13    "down_block_types": [14      "CrossAttnDownBlock2D",15      "CrossAttnDownBlock2D",16      "CrossAttnDownBlock2D"17    ],18    "downsample_padding": 1,19    "flip_sin_to_cos": true,20    "freq_shift": 0,21    "in_channels": 4,22    "layers_per_block": 1,23    "mid_block_type": null,24    "norm_eps": 1e-05,25    "norm_num_groups": 32,26    "out_channels": 4,27    "sample_size": 64,28    "up_block_types": [29      "CrossAttnUpBlock2D",30      "CrossAttnUpBlock2D",31      "CrossAttnUpBlock2D"32    ]33  }