ps: float = 1e-05
resnet_time_scale_shift: str = 'default'
add_attention: bool = True
class_embed_type: typing.Optional[str] = None
num_class_embeds: typing.Optional[int] = None

)


Parameters 

sample_size (int or Tuple[int, int], optional, defaults to None) —
Height and width of input/output sample.


in_channels (int, optional, defaults to 3) — Number of channels in the input image.


out_channels (int, optional, defaults to 3) — Number of 