honk-140k / unet /config.json
patrickvonplaten's picture
Update unet/config.json
6f74bc2
{
"_class_name": "UNet1DModel",
"_diffusers_version": "0.5.0.dev0",
"block_out_channels": [
128,
128,
256,
256,
512,
512,
512,
512,
512,
512,
512,
512,
512
],
"down_block_types": [
"DownBlock1DNoSkip",
"DownBlock1D",
"DownBlock1D",
"DownBlock1D",
"DownBlock1D",
"DownBlock1D",
"DownBlock1D",
"DownBlock1D",
"AttnDownBlock1D",
"AttnDownBlock1D",
"AttnDownBlock1D",
"AttnDownBlock1D",
"AttnDownBlock1D"
],
"flip_sin_to_cos": true,
"freq_shift": 0,
"in_channels": 2,
"mid_block_type": "UNetMidBlock1D",
"extra_in_channels": 16,
"out_channels": 2,
"sample_rate": 16000,
"sample_size": 65536,
"time_embedding_type": "fourier",
"up_block_types": [
"AttnUpBlock1D",
"AttnUpBlock1D",
"AttnUpBlock1D",
"AttnUpBlock1D",
"AttnUpBlock1D",
"UpBlock1D",
"UpBlock1D",
"UpBlock1D",
"UpBlock1D",
"UpBlock1D",
"UpBlock1D",
"UpBlock1D",
"UpBlock1DNoSkip"
],
"use_timestep_embedding": false
}