File size: 1,065 Bytes
901e9b3 80fd2c9 f80e5ef 901e9b3 80fd2c9 901e9b3 |
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
{
"_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
}
|