mple (torch.FloatTensor of shape (batch_size, num_channels, sample_size)) —
Hidden states output. Output of last layer of model.




UNet1DModel
	

class diffusers.UNet1DModel

<
source
>
(
sample_size: int = 65536
sample_rate: typing.Optional[int] = None
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 =