h.FloatTensor) — (batch, sequence_length, feature_dim) encoder hidden states


return_dict (bool, optional, defaults to True) —
Whether or not to return a models.unet_2d_condition.UNet2DConditionOutput instead of a plain tuple.


Returns

UNet2DConditionOutput or tuple



UNet2DConditionOutput if return_dict is True, otherwise a tuple. When
returning a tuple, the first element is the sample tensor.



set_attention_slice

<
source
>
(
slice_siz