bglick13's picture
Update unet/config.json
0fac8a0
raw
history blame
316 Bytes
{"down_block_types": ["DownResnetBlock1D", "DownResnetBlock1D", "DownResnetBlock1D", "DownResnetBlock1D"], "block_out_channels": [32, 64, 128, 256], "up_block_types": ["UpResnetBlock1D", "UpResnetBlock1D", "UpResnetBlock1D"], "layers_per_block": 1, "use_timestep_embedding": true, "out_block_type": "OutConv1DBlock"}