File size: 792 Bytes
06d2c78 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | {
"_class_name": "UNet1DModel",
"_diffusers_version": "0.37.1",
"act_fn": "silu",
"block_out_channels": [
64,
128,
256,
512
],
"down_block_types": [
"DownBlock1DNoSkip",
"DownBlock1D",
"AttnDownBlock1D",
"AttnDownBlock1D"
],
"downsample_each_block": false,
"extra_in_channels": 128,
"flip_sin_to_cos": true,
"freq_shift": 0.0,
"in_channels": 63,
"layers_per_block": 2,
"mid_block_type": "UNetMidBlock1D",
"norm_num_groups": 8,
"out_block_type": null,
"out_channels": 21,
"sample_rate": null,
"sample_size": 256,
"time_embedding_dim": null,
"time_embedding_type": "fourier",
"up_block_types": [
"AttnUpBlock1D",
"AttnUpBlock1D",
"UpBlock1D",
"UpBlock1DNoSkip"
],
"use_timestep_embedding": false
}
|