um_train_timesteps: Optional = None  )   Parameters   sample_size (int or Tuple[int, int], optional, defaults to None) —
Height and width of input/output sample. Dimensions must be a multiple of 2 ** (len(block_out_channels) - 1).   in_channels (int, optional, defaults to 3) — Number of channels in the input sample.   out_channels (int, optional, defaults to 3) — Number of channels in the output.   center_input_sample (bool, optional, defaults 