ck (int, optional, defaults to 1) — added number of layers in a UNet block.


downsample_each_block (int, optional, defaults to False —
experimental feature for using a UNet without upsampling.



UNet1DModel is a 1D UNet model that takes in a noisy sample and a timestep and returns sample shaped output.
This model inherits from ModelMixin. Check the superclass documentation for the generic methods the library
implements for all the model (such