esidual of the middle unet block.   return_dict (bool, optional, defaults to True) —
Whether or not to return a UNet3DConditionOutput instead of a plain
tuple.   Returns 
UNet3DConditionOutput or tuple
  
If return_dict is True, an 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