File size: 707 Bytes
054b79b |
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 |
{
"_class_name": "UNet1DModel",
"_diffusers_version": "0.12.1",
"act_fn": "mish",
"block_out_channels": [
32,
64,
128,
256
],
"down_block_types": [
"DownResnetBlock1D",
"DownResnetBlock1D",
"DownResnetBlock1D",
"DownResnetBlock1D"
],
"downsample_each_block": true,
"extra_in_channels": 0,
"flip_sin_to_cos": false,
"freq_shift": 1,
"in_channels": 14,
"layers_per_block": 1,
"mid_block_type": "ValueFunctionMidBlock1D",
"norm_num_groups": 8,
"out_block_type": "ValueFunction",
"out_channels": 14,
"sample_rate": null,
"sample_size": 65536,
"time_embedding_type": "positional",
"up_block_types": [],
"use_timestep_embedding": true
}
|