pub struct UNet2DConditionModel { /* private fields */ }
Implementations§
source§impl UNet2DConditionModel
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§
Auto Trait Implementations§
impl Freeze for UNet2DConditionModel
impl !RefUnwindSafe for UNet2DConditionModel
impl Send for UNet2DConditionModel
impl Sync for UNet2DConditionModel
impl Unpin for UNet2DConditionModel
impl !UnwindSafe for UNet2DConditionModel
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more