f 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 UNetMotionModel forward method.  freeze_unet2d_params  < source > ( )    Freeze the weights of just the UNet2DConditionModel, and leave the motion modules
unfrozen for fine tuning.  fuse_qkv_projections  < source > ( )    Enables fused QKV projections. For self-attention modules