pub struct TwoWayTransformerWeights {
pub layers: Vec<TwoWayAttentionBlockWeights>,
pub final_attn_token_to_image: AttentionWeights,
pub norm_final_g: Vec<f32>,
pub norm_final_b: Vec<f32>,
pub embed_dim: usize,
}Fields§
§layers: Vec<TwoWayAttentionBlockWeights>§final_attn_token_to_image: AttentionWeights§norm_final_g: Vec<f32>§norm_final_b: Vec<f32>§embed_dim: usizeAuto Trait Implementations§
impl Freeze for TwoWayTransformerWeights
impl RefUnwindSafe for TwoWayTransformerWeights
impl Send for TwoWayTransformerWeights
impl Sync for TwoWayTransformerWeights
impl Unpin for TwoWayTransformerWeights
impl UnsafeUnpin for TwoWayTransformerWeights
impl UnwindSafe for TwoWayTransformerWeights
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more