TransformerOutput instead of a plain
tuple.   Returns 
~models.prior_transformer.PriorTransformerOutput or tuple
  
If return_dict is True, a ~models.prior_transformer.PriorTransformerOutput is returned, otherwise a
tuple is returned where the first element is the sample tensor.
  The PriorTransformer forward method.  set_attn_processor  < source > ( processor: Union  )   Parameters   processor (dict of AttentionProcessor or only AttentionProce