one
in_channels: int = 2
out_channels: int = 2
extra_in_channels: int = 0
time_embedding_type: str = 'fourier'
flip_sin_to_cos: bool = True
use_timestep_embedding: bool = False
freq_shift: float = 0.0
down_block_types: typing.Tuple[str] = ('DownBlock1DNoSkip', 'DownBlock1D', 'AttnDownBlock1D')
up_block_types: typing.Tuple[str] = ('AttnUpBlock1D', 'UpBlock1D', 'UpBlock1DNoSkip')
mid_block_type: typing.Tuple[str] = 'UNetMidBlock1D'
out_block_type