hat if specified is passed along to the AttnProcessor.   Returns 
~models.unet_3d_condition.UNet3DConditionOutput or tuple
  
If return_dict is True, an ~models.unet_3d_condition.UNet3DConditionOutput is returned, otherwise
a tuple is returned where the first element is the sample tensor.
  The UNet3DConditionModel forward method.  set_attention_slice  < source > ( slice_size: Union  )   Parameters   slice_size (str or int or list(int), optiona