Struct UNet2DConditionModel 

pub struct UNet2DConditionModel { /* private fields */ }

Implementations

impl UNet2DConditionModel

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

pub fn forward( &self, xs: &Tensor, timestep: f64, encoder_hidden_states: &Tensor, ) -> Result<Tensor>

pub fn forward_with_additional_residuals( &self, xs: &Tensor, timestep: f64, encoder_hidden_states: &Tensor, down_block_additional_residuals: Option<&[Tensor]>, mid_block_additional_residual: Option<&Tensor>, ) -> Result<Tensor>

Trait Implementations

impl Debug for UNet2DConditionModel

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

Formats the value using the given formatter.