Struct CrossAttnUpBlock2D 

pub struct CrossAttnUpBlock2D {
    pub upblock: UpBlock2D,
    pub attentions: Vec<SpatialTransformer>,
    pub config: CrossAttnUpBlock2DConfig,
    /* private fields */
}

Fields

upblock: UpBlock2D

attentions: Vec<SpatialTransformer>

config: CrossAttnUpBlock2DConfig

Implementations

impl CrossAttnUpBlock2D

pub fn new( vs: VarBuilder<'_>, in_channels: usize, prev_output_channels: usize, out_channels: usize, temb_channels: Option<usize>, use_flash_attn: bool, config: CrossAttnUpBlock2DConfig, ) -> Result<Self>

pub fn forward( &self, xs: &Tensor, res_xs: &[Tensor], temb: Option<&Tensor>, upsample_size: Option<(usize, usize)>, encoder_hidden_states: Option<&Tensor>, ) -> Result<Tensor>

Trait Implementations

impl Debug for CrossAttnUpBlock2D

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.