 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 as downloading or saving, etc.)

forward

<
source
>
(
sample: Floa