Nathan Lambert
update mid and out block
ff13ce2
{
"_class_name": "UNet1DModel",
"_diffusers_version": "0.5.0.dev0",
"_name_or_path": "fusing/ddpm-unet-rl-hopper-hor128",
"act_fn": "mish",
"always_downsample": false,
"block_out_channels": [
32,
128,
256
],
"down_block_types": [
"DownResnetBlock1D",
"DownResnetBlock1D",
"DownResnetBlock1D"
],
"in_channels": 14,
"layers_per_block": 1,
"mid_block_type": "MidResTemporalBlock1D",
"norm_num_groups": 8,
"out_block_type": "OutConv1DBlock",
"out_channels": 14,
"up_block_types": [
"UpResnetBlock1D",
"UpResnetBlock1D"
]
}