pub struct CrossAttnDownBlock2D {
pub config: CrossAttnDownBlock2DConfig,
/* private fields */
}
Fields§
§config: CrossAttnDownBlock2DConfig
Implementations§
source§impl CrossAttnDownBlock2D
impl CrossAttnDownBlock2D
pub fn new( vs: VarBuilder<'_>, in_channels: usize, out_channels: usize, temb_channels: Option<usize>, use_flash_attn: bool, config: CrossAttnDownBlock2DConfig ) -> Result<Self>
pub fn forward( &self, xs: &Tensor, temb: Option<&Tensor>, encoder_hidden_states: Option<&Tensor> ) -> Result<(Tensor, Vec<Tensor>)>
Trait Implementations§
Auto Trait Implementations§
impl Freeze for CrossAttnDownBlock2D
impl !RefUnwindSafe for CrossAttnDownBlock2D
impl Send for CrossAttnDownBlock2D
impl Sync for CrossAttnDownBlock2D
impl Unpin for CrossAttnDownBlock2D
impl !UnwindSafe for CrossAttnDownBlock2D
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